結果

問題 No.1916 Making Palindrome on Gird
ユーザー wolgnik
提出日時 2022-04-29 22:02:37
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 1,306 bytes
コンパイル時間 168 ms
コンパイル使用メモリ 82,032 KB
実行使用メモリ 847,940 KB
最終ジャッジ日時 2024-06-29 03:29:17
合計ジャッジ時間 2,777 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 10 MLE * 1 -- * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline
H, W = map(int, input().split())
a = [list(input())[: -1] for _ in range(H)]
mod = 10 ** 9 + 7

if a[0][0] != a[-1][-1]:
  print(0)
  exit(0)

from collections import deque as dq
Q = dq([(0, H * W - 1)])

d1 = [1, 0]
d2 = [-1, 0]

dp = [[0] * (H * W) for _ in range(H * W)]
dp[0][-1] = 1
vis = [[0] * (H * W) for _ in range(H * W)]

while len(Q):
  u, v = Q.popleft()
  if vis[u][v]: continue
  vis[u][v] = 1

  i, j = u // W, u % W
  y, x = v // W, v % W

  for k in range(2):
    i2, j2 = i + d1[k], j + d1[~k]
    if not (i2 in range(H) and j2 in range(W)): continue
    u2 = i2 * W + j2
    for kk in range(2):
      y2, x2 = y + d2[kk], x + d2[~kk]
      if not (y2 in range(H) and x2 in range(W)): continue
      if a[i2][j2] != a[y2][x2]: continue
      v2 = y2 * W + x2

      dp[u2][v2] += dp[u][v]
      dp[u2][v2] %= mod
      Q.append((u2, v2))
      #print((i, j), (y,x),(i2, j2),(y2,x2))

  #print((i, j), (y, x), len(Q))

res = 0
for i in range(H):
  for j in range(W):
    res += dp[i * W + j][i * W + j]
    res %= mod
for i in range(H):
  for j in range(W):
    for k in range(2):
      u, v = i + d1[k], j + d1[~k]
      if u in range(H) and v in range(W) and a[i][j] == a[u][v]:
        res += dp[i * W + j][u * W + v]
        res %= mod
print(res)
0