本文已参与「新人创作礼」活动,一起开启掘金创作之路
本人为 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 = ' ') # 因为记录的是索引位置,所以输出时要加一。