【PTA-python】第4章-23 求矩阵的局部极大值

分析

定义一个判断部分最大值的函数：大于四周的函数，就返回True
对非边界值的判断:范围是range(1,n-1)
Python中append和extend的区别

list.append(object) 向列表中添加一个对象object
list.extend(sequence) 把一个序列seq的内容添加到列表中

改进

如果使用a.appen()，那么大列表里的元素是小列表。

a.append(list(map(int,input().split())))

那么就可以直接通过i,j访问特定行列下标：

if a[i][j]>a[i][j-1] and a[i][j]>a[i][j+1] and a[i][j]>a[i+1][j] and a[i][j]>a[i-1][j]:

题目

给定M行N列的整数矩阵A，如果A的非边界元素A[i][j]大于相邻的上下左右4个元素，那么就称元素A[i][j]是矩阵的局部极大值。本题要求给定矩阵的全部局部极大值及其所在的位置。
输入格式：

输入在第一行中给出矩阵A的行数M和列数N（3≤M,N≤20）；最后M行，每行给出A在该行的N个元素的值。数字间以空格分隔。
输出格式：

每行按照“元素值 行号 列号”的格式输出一个局部极大值，其中行、列编号从1开始。要求按照行号递增输出；若同行有超过1个局部极大值，则该行按列号递增输出。若没有局部极大值，则输出“None 总行数 总列数”。

解法

def partmax(column,m,n,lst=[]):
    if lst[m*column+n]>lst[m*column+n+1] and lst[m*column+n]>lst[m*column+n-1] and lst[m*column+n]>lst[(m-1)*column+n] and lst[m*column+n]>lst[(m+1)*column+n]:
        return True
    return False
m,n=map(int,input().split())
lst=[]
cnt=0
for i in range(m):
    lst.extend(list(map(int,input().split())))
for i in range(1,m-1):
    for j in range(1,n-1):
        if partmax(n,i,j,lst)==True:
            cnt+=1
            print("{} {} {}".format(lst[i*n+j],i+1,j+1))#i,j和实际的行列下标相差1，输出需要加上1
if cnt==0:
    print("None {} {}".format(m,n))