多维数组:寻找鞍点

208 阅读2分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路

本人为 python 小白,写这题的代码有两个原因。一是:我第一次做这道题没有什么思路,但百度到的都是 C++ 的解法,所以打算写一个 python 的代码。二是:我打算把思路过程写下来,这样我以后也可以更好理解自己当时的想法。 (由于笔者接触 python 不久,所以写的代码有些冗长而且不知道如何优化,还望阅者谅解。如有错误,恳请指正。)

题目描述:

给定一个5*5的矩阵,每行只有一个最大值,每列只有一个最小值,寻找这个矩阵的鞍点。 鞍点指的是矩阵中的一个元素,它是所在行的最大值,并且是所在列的最小值。 例如:在下面的例子中(第4行第1列的元素就是鞍点,值为8 )。 11 3 5 6 9 12 4 7 8 10 10 5 6 9 11 8 6 4 7 2 15 10 11 20 25

输入格式:

输入包含一个5行5列的矩阵

输出格式:

如果存在鞍点,输出鞍点所在的行、列及其值,如果不存在,输出"not found"

输入样例:

11 3 5 6 9 12 4 7 8 10 10 5 6 9 11 8 6 4 7 2 15 10 11 20 25

输出样例:

4 1 8

简单的思路分析

这道题的思路比较容易想到,就是找到某一行的最大值,然后判断这个最大值是否为它所在的列的最小值。

代码实现

因为自学的缘故,我总是用 j 表示行,用 i 表示列

grid = []

count = 0
while count < 5:
    list1 = []
    temp = input().split()
    for i in temp:
        list1.append(int(i))
    grid.append(list1)
    count += 1

j_1 = []
i_1 = []

for i in range(len(grid)):
	# 找出每一行的最大值以及它在这一行的位置,这个位置就是最大值所在的列的位置。
    max_num = (max(grid[i]))
    index = grid[i].index(max_num) # 这是列的位置索引
    list1 = []
    # 找出最大值所在列的最小值并记录它在那一行
    for j in range(len(grid)):
        list1.append(grid[j][index])
    min_num = min(list1)
    index1 = list1.index(max_num) # 这是行的位置索引
    # 判断最大值是否等于最小值,如果相等记录鞍点的位置
    if min_num == max_num:
        j_1.append(index1)
        i_1.append(index)
        
if len(i_1) == 0:
    print("not found")
else:
    for i in range(len(j_1)):
        print(j_1[i] + 1, i_1[i] + 1, grid[j_1[i]][i_1[i]], sep = ' ') # 因为记录的是索引位置,所以输出时要加一。