国际空间研究组织 | ISRO CS 2009 | 问题 5

题目描述

对于一个n x n的矩阵，找到所有行和列的总和相等的位置，即矩阵的“幸运数字”。

输入格式：

• 第一行包括一个整数T，表示测试用例的数量。接下来的T行包含：
• 每个测试用例的第一行包括一个整数n，表示矩阵的维数。
• 然后有n行，每行包括n个空格分隔的整数，表示矩阵中的元素，这些元素可以是正的也可以是负的。

输出格式：

• 对于每个测试用例，以以下格式输出幸运数字的行和列位置：
• 如果没有“幸运数字”，请打印“未找到”。
• 如果有一个“幸运数字”，请把它的行和列位置打印出来。
• 如果有多个“幸运数字”，请选择任意一个并打印其行和列位置。

例子

输入：
2
4
1 2 3 4
2 3 4 1
3 4 1 2
4 1 2 3
3
1 2 3
2 3 1
3 1 2

输出：
Lucky number 7 found at (2,2)
未找到

思路

可以遍历整个矩阵，然后统计每一行的总和及每一列的总和。当行总和与列总和相等时，说明该位置为幸运数字。

代码

def find_lucky_number(matrix):
    n = len(matrix)
    rows_sum = [0] * n
    cols_sum = [0] * n
    
    # 统计每行每列的和
    for i in range(n):
        for j in range(n):
            rows_sum[i] += matrix[i][j]
            cols_sum[j] += matrix[i][j]
    
    # 遍历整个矩阵，查找幸运数字
    for i in range(n):
        for j in range(n):
            if rows_sum[i] == cols_sum[j]:
                return f"Lucky number {rows_sum[i]} found at ({i+1},{j+1})"
    return "未找到"

返回的代码片段中包括一个名为 find_lucky_number 的函数，可以传入一个二维数组 matrix，返回一个字符串，表示幸运数字的行和列位置。如果没有幸运数字，则返回 "未找到"。