I15-leetcode1878矩阵中最大的三个菱形和

题目描述：

提示：

思路与算法：

需要注意单独的一个格子也是菱形。

代码：

class Answer:#维护三个最大值
    def __init__(self):
        self.ans=[0,0,0]
        
    def put(self,x):#ans[0]>ans[1]>ans[2]
        _ans=self.ans#浅拷贝
        if x>_ans[0]:_ans[0],_ans[1],_ans[2]=x,_ans[0],_ans[1]
        elif x!=_ans[0] and x>_ans[1]:_ans[1],_ans[2]=x,_ans[1]
        elif x != _ans[0] and x != _ans[1] and x > _ans[2]:_ans[2]=x
        
    def get(self):
        _ans=self.ans
        return [num for num in _ans if num!=0]#可能不到三个

        
def fun(grid):
    m,n=len(grid),len(grid[0])
    if m==1 and n==1:return grid[0]#只有一个数字
    sum1=[[0]*(n+2) for _ in range(m+1)]
    sum2=[[0]*(n+2) for _ in range(m+1)]
    for i in range(1,m+1):
        for j in range(1,n+1):
            sum1[i][j]=sum1[i-1][j-1]+grid[i-1][j-1]#左上
            sum2[i][j]=sum2[i-1][j+1]+grid[i-1][j-1]#右上
    ans=Answer()
    for i in range(m):
        for j in range(n):
            ans.put(grid[i][j])#单独一个格子也是菱形
            for k in range(i+2,m,2):
                ux,uy=i,j #上
                dx,dy=k,j #下
                lx,ly=(i+k)//2,j-(k-i)//2 #左
                rx,ry=(i+k)//2,j+(k-i)//2 #右
                if ly<0 or ry>=n:break#超出边界

                ans.put(
                    (sum2[lx + 1][ly + 1] - sum2[ux][uy + 2]) +
                    (sum1[rx + 1][ry + 1] - sum1[ux][uy]) +
                    (sum1[dx + 1][dy + 1] - sum1[lx][ly]) +
                    (sum2[dx + 1][dy + 1] - sum2[rx][ry + 2]) -
                    (grid[ux][uy] + grid[dx][dy] + grid[lx][ly] + grid[rx][ry])
                    )
    return ans.get()
        
#测试
grid = [[1,2,3],[4,5,6],[7,8,9]]
print(fun(grid))