推 rainkaras: 寶 你最棒 08/10 00:59
→ sixB: 你是最棒的 08/10 01:07
亂七八糟
跟我的人生依樣
一二三四五
def numMagicSquaresInside(self, grid: List[List[int]]) -> int:
m, n = len(grid), len(grid[0])
ans = 0
for i in range(m):
for j in range(n):
if i+2<m and j+2<n:
cnt = set()
for ii in range(i,i+3):
for jj in range(j,j+3):
if grid[ii][jj] <= 9:
cnt.add(grid[ii][jj])
if len(cnt) != 9:
continue
sums = []
# row
for k in range(3):
sums.append(grid[i+k][j]+grid[i+k][j+1]+grid[i+k][j+2])
# col
for k in range(3):
sums.append(grid[i][j+k]+grid[i+1][j+k]+grid[i+2][j+k])
# diagonal
sums.append(grid[i][j]+grid[i+1][j+1]+grid[i+2][j+2])
sums.append(grid[i][j+2]+grid[i+1][j+1]+grid[i+2][j])
if all([s==15 for s in sums]):
ans += 1
return ans
--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 125.228.146.144 (臺灣)
※ 文章網址: https://www.ptt.cc/bbs/Marginalman/M.1723222656.A.FAC.html
※ 編輯: DJYOMIYAHINA (125.228.146.144 臺灣), 08/10/2024 00:58:07