关于嵌套循环定位n×n矩阵最大值位置的技术疑问

嘿，这个需求其实很常见，我来一步步帮你搞定！首先你已经找对了获取全局最大值的方向，不过咱们可以先优化一下最大值的获取方式，再搞定循环终止的问题。

第一步：高效获取矩阵最大值 #

你用的max(sum(matrix, []))确实能拿到最大值，但如果矩阵很大的话，sum(matrix, [])会把整个矩阵拼成一个新列表，有点浪费内存。更高效的方式是用生成器表达式：

max_val = max(num for row in matrix for num in row)

它会逐个遍历矩阵元素，不用额外创建大列表，性能更好～

第二步：嵌套循环中找到最大值后终止循环 #

这里有两种常用的方法，看你喜欢哪种：

方法1：用标志变量控制循环

设置一个布尔变量标记是否找到最大值，找到后先跳出内层循环，再在外层循环里检查标志，跳出外层循环：

matrix = [
    [1, 3, 5],
    [2, 8, 4],
    [7, 6, 9]
]

max_val = max(num for row in matrix for num in row)
max_position = None
found = False  # 标志位：是否找到最大值

# 遍历行，i是行索引
for i, row in enumerate(matrix):
    if found:
        break  # 已经找到，直接跳出外层循环
    # 遍历列，j是列索引
    for j, num in enumerate(row):
        if num == max_val:
            max_position = (i, j)
            found = True
            break  # 找到后跳出内层循环

print(f"最大值的位置是：行{i+1}，列{j+1}" if max_position else "矩阵为空")

方法2：把逻辑封装成函数，用return直接终止

这种方式更简洁，找到最大值后直接return位置，所有循环都会自动终止：

def find_max_pos(matrix):
    if not matrix:
        return None  # 处理空矩阵的情况
    max_val = max(num for row in matrix for num in row)
    for i, row in enumerate(matrix):
        for j, num in enumerate(row):
            if num == max_val:
                return (i, j)  # 找到就返回，直接终止所有循环

# 测试用例
matrix = [
    [1, 3, 5],
    [2, 8, 4],
    [7, 6, 9]
]

pos = find_max_pos(matrix)
if pos:
    print(f"最大值的位置是（行索引：{pos[0]}，列索引：{pos[1]}）")
    # 如果习惯从1开始计数的话：
    # print(f"最大值的位置是（行：{pos[0]+1}，列：{pos[1]+1}）")
else:
    print("矩阵为空，没有最大值")

补充说明 #

• 如果矩阵中有多个相同的最大值，上面的代码会返回第一个出现的位置；如果需要收集所有最大值的位置，就去掉终止循环的逻辑，把位置添加到列表里就行。
• enumerate()函数用来同时获取索引和元素，比手动维护计数器更方便。

内容的提问来源于stack exchange，提问作者AAZA