結果

問題 No.866 レベルKの正方形
ユーザー vwxyz
提出日時 2024-04-14 14:21:57
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
MLE  
実行時間 -
コード長 708 bytes
コンパイル時間 761 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 821,248 KB
最終ジャッジ日時 2024-10-03 13:18:39
合計ジャッジ時間 4,082 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 3
other AC * 8 MLE * 1 -- * 13
権限があれば一括ダウンロードができます

ソースコード

diff # 

H,W,K=map(int,input().split())
A=[[ord(a)-97 for a in input()] for h in range(H)]
inf=1<<30
dp=[[inf]*(H*W) for a in range(26)]
ans=0
for h in range(H):
    for w in range(W):
        for a in range(26):
            if A[h][w]==a:
                dp[a][h*W+w]=0
            if h:
                dp[a][h*W+w]=min(dp[a][h*W+w],dp[a][(h-1)*W+w]+1)
            if w:
                dp[a][h*W+w]=min(dp[a][h*W+w],dp[a][h*W+w-1]+1)
            if h and w:
                dp[a][h*W+w]=min(dp[a][h*W+w],dp[a][(h-1)*W+w-1]+1)
        idx=[dp[a][h*W+w] for a in range(26) if dp[a][h*W+w]<=min(h,w)]
        idx.sort()
        idx.append(min(h,w)+1)
        if len(idx)>K:
            ans+=idx[K]-idx[K-1]
print(ans)
0