PTA 输出鞍点的行列下标

问题描述：

一个矩阵元素的“鞍点”是指该位置上的元素值在该行上最大、在该列上最小。

本题要求编写程序，求一个给定的n阶方阵的鞍点的行列值。

输入格式:

输入第一行给出一个正整数n（1≤n≤6），随后n行，每行给出n个整数，其间以空格分隔。

输出格式:

输出在一行中按照“行下标 列下标”（下标从0开始）的格式输出鞍点的位置。如果鞍点不存在，则输出“NONE”。题目保证给出的矩阵至多存在一个鞍点。

输入样例1:

4
1 9 4 1
4 7 3 6
1 9 3 2
0 8 8 9

输出样例1:

1 1

输入样例2:

4
7   7  4  1
14  8  3  6
6   6  7  2
10  7  8  9

输出样例2:

NONE

代码实现 

def find(matrix):
    n = len(matrix)
    for i in range(n):
        row_max = max(matrix[i])
        for j in range(n):
            if matrix[i][j] == row_max:
                col_min = min([matrix[k][j] for k in range(n)])
                if matrix[i][j] == col_min:
                    return [i, j]
    return "NONE"

n = int(input())
matrix = []
for _ in range(n):
    row = list(map(int, input().split()))
    matrix.append(row)

result = find(matrix)
if result == "NONE":
    print(result)
else:
    print(result[0], result[1])

思路解析：

        题目要求找到矩阵中的鞍点，即在该位置上元素值是该行最大且该列最小的。我们可以遍历矩阵的每个元素，分别判断是否满足这两个条件。

        具体实现时，我们先遍历每一行，找到当前行的最大值row_max，然后在该行中继续遍历每个元素。对于每个元素，我们判断其是否等于row_max，如果相等，则继续在该列中找到最小值col_min。如果该元素也等于col_min，则说明找到了鞍点，返回该位置的行列下标。

如果遍历结束后仍未找到鞍点，我们返回字符串"NONE"表示鞍点不存在。

        在主程序中，我们首先读入输入的矩阵大小n，然后逐行读入矩阵的每一行，并将其存储在一个二维列表matrix中。最后，调用find_saddle_point()函数找到鞍点，并按照题目要求输出结果。