結果

問題 No.866 レベルKの正方形
コンテスト
ユーザー noisy_noimin
提出日時 2019-08-17 15:35:44
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,173 bytes
コンパイル時間 626 ms
コンパイル使用メモリ 70,488 KB
実行使用メモリ 425,940 KB
最終ジャッジ日時 2024-09-24 19:53:07
合計ジャッジ時間 29,622 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20 WA * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <algorithm>

using namespace std;

using ll = long long;

int h,w,k;
int a[2000][2000], r[2001][2001][26];

int main() {
    cin >> h >> w >> k;

    char c;
    for(int i=0;i<h;++i) {
        for(int j=0;j<w;++j) {
            cin >> c;
            a[i][j] = int(c - 'a');
        }
    }
    fill_n(r[0][0], 2001*2001*26, 1<<30);
    for(int i=h-1;i>=0;--i) {
        for(int j=w-1;j>=0;--j) {
            for(int ci=0;ci<26;++ci) {
                if(a[i][j] == ci) {
                    r[i][j][ci] = 1;
                } else {
                    r[i][j][ci] = min(1<<30, min({r[i+1][j][ci], r[i][j+1][ci], r[i+1][j+1][ci]}) + 1);
                    if(r[i][j][ci] > min(h-i, w-j)) r[i][j][ci] = 1<<30;
                }
            }
        }
    }

    ll ans = 0;
    for(int i=0;i<h;++i) {
        for(int j=0;j<w;++j) {
            sort(r[i][j], r[i][j]+26);
            if(r[i][j][k-1] == 1<<30) continue;
            if(r[i][j][k] == 1<<30) {
                ans += min(h-i, w-j) + 1 - r[i][j][k-1];
            } else {
                ans += r[i][j][k] - r[i][j][k-1];
            }
        }
    }

    cout << ans << endl;
}
0