結果

問題 No.1916 Making Palindrome on Gird
コンテスト
ユーザー roaris
提出日時 2022-05-14 19:37:14
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,318 ms / 3,000 ms
コード長 1,542 bytes
コンパイル時間 583 ms
コンパイル使用メモリ 82,356 KB
実行使用メモリ 148,608 KB
最終ジャッジ日時 2024-07-23 06:11:14
合計ジャッジ時間 19,027 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline
from collections import *

H, W = map(int, input().split())
S = [input()[:-1] for _ in range(H)]

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

ind = [[-1]*W for _ in range(H)]
d = defaultdict(list)

for i in range(H):
    for j in range(W):
        ind[i][j] = len(d[H-1-i+W-1-j])
        d[H-1-i+W-1-j].append((i, j))

dp = [[[0]*210 for _ in range(W)] for _ in range(H)]
dp[0][0][0] = 1
MOD = 10**9+7

for i in range(H):
    for j in range(W):
        for k in range(210):
            if k>=len(d[i+j]):
                continue
            
            i2, j2 = d[i+j][k]
            
            if i>=i2 and j>=j2:
                continue
            
            for ni, nj in [(i+1, j), (i, j+1)]:
                for ni2, nj2 in [(i2-1, j2), (i2, j2-1)]:
                    if not (0<=ni<H and 0<=nj<W and 0<=ni2<H and 0<=nj2<W):
                        continue
                    
                    if (ni, nj)==(ni2, nj2) or S[ni][nj]==S[ni2][nj2]:
                        dp[ni][nj][ind[ni2][nj2]] += dp[i][j][k]
                        dp[ni][nj][ind[ni2][nj2]] %= MOD

ans = 0

for i in range(H):
    for j in range(W):
        if 2*(i+j)==H+W-2:
            ans += dp[i][j][ind[i][j]]
            ans %= MOD
            
        for ni, nj in [(i+1, j), (i, j+1)]:
            if not (0<=ni<H and 0<=nj<W):
                continue
            
            if i+j==H-ni+W-nj-2 and i+j+H-ni+W-nj==H+W-1:
                ans += dp[i][j][ind[ni][nj]]
                ans %= MOD

print(ans)
0