結果

問題 No.866 レベルKの正方形
コンテスト
ユーザー LyricalMaestro
提出日時 2026-05-05 18:07:15
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
MLE  
実行時間 -
コード長 1,689 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 628 ms
コンパイル使用メモリ 84,864 KB
実行使用メモリ 1,100,952 KB
最終ジャッジ日時 2026-05-05 18:07:54
合計ジャッジ時間 8,904 ms
ジャッジサーバーID
（参考情報） 		judge2_0 / judge3_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 3
other AC * 8 MLE * 1 -- * 13
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

## https://yukicoder.me/problems/no/866

def solve2(h, w, K, l, cum_dp_array):
    h1 = h + l - 1
    w1 = w + l - 1

    answer = 0
    for word in range(26):
        cum_dp = cum_dp_array[word]
        ans = cum_dp[h1 + 1][w1 + 1] - cum_dp[h1 + 1][w] - cum_dp[h][w1 +1] + cum_dp[h][w]
        if ans > 0:
            answer += 1
    return answer <= K

def solve(H, W, K, cum_dp_array):
    answer = 0
    for h in range(H):
        for w in range(W):
            low = 1
            high = min( (H - h), (W - w))
            while high - low > 1:
                mid = (high + low) // 2
                if solve2(h, w, K, mid, cum_dp_array):
                    low = mid
                else:
                    high = mid
            if solve2(h, w, K, high, cum_dp_array):
                answer += high
            else:
                answer += low
    return answer


def main():
    H, W, K = map(int, input().split())
    C = []
    for _ in range(H):
        C.append(input())

    cum_dp_array = []
    for word in range(26):
        dp = [[0] * W for _ in range(H)]
        for h in range(H):
            for w in range(W):
                if ord(C[h][w]) - ord("a") == word:
                    dp[h][w] += 1

        cum_dp = [[0] * (W + 1 ) for _ in range(H + 1)]
        for h in range(H):
            row  = 0
            for w in range(W):
                row += dp[h][w]
                cum_dp[h + 1][w + 1] = cum_dp[h][w + 1] + row
        cum_dp_array.append(cum_dp)

    ans1 = solve(H, W, K, cum_dp_array)        
    if K > 1:
        ans2 = solve(H, W, K - 1, cum_dp_array)
        ans1 -= ans2
    print(ans1)












if __name__ == "__main__":
    main()
0