题目描述:
提示:
思路与算法:
需要注意单独的一个格子也是菱形。
代码:
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))