結果

問題 No.866 レベルKの正方形
コンテスト
ユーザー kakira9618
提出日時 2019-08-17 00:52:53
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
MLE  
実行時間 -
コード長 2,716 bytes
コンパイル時間 1,989 ms
コンパイル使用メモリ 178,640 KB
実行使用メモリ 813,824 KB
最終ジャッジ日時 2024-09-24 07:18:08
合計ジャッジ時間 5,394 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 3
other AC * 8 MLE * 1 -- * 13
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;
#define all(x) (x).begin(), (x).end()
#define rep(i, n) for (int i = 0; i < (n); i++)
#define endl "\n"
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;

template <typename T> ostream &operator<<(ostream &os, const vector<T> &vec){os << "["; for (const auto &v : vec) {os << v << ","; } os << "]"; return os; }
template <typename T, typename U> ostream &operator<<(ostream &os, const pair<T, U> &p) {os << "(" << p.first << ", " << p.second << ")"; return os;}

void solve() {
    int H, W, K;
    cin >> H >> W >> K;
    vector<string> C(H);
    for (int i = 0; i < H; i++) {
        cin >> C[i];
    }
    vector<vector<vector<int>>> flag(26, vector<vector<int>>(H, vector<int>(W, 0)));
    vector<vector<vector<int>>> acc(26, vector<vector<int>>(H + 1, vector<int>(W + 1, 0)));

    for (int c = 0; c < 26; c++) {
        for (int i = 0; i < H; i++) {
            for (int j = 0; j < W; j++) {
                if (C[i][j] == 'a' + c) {
                    flag[c][i][j] = 1;
                }
            }
        }
        for (int i = 0; i < H; i++) {
            for (int j = 0; j < W; j++) {
                acc[c][i + 1][j + 1] = acc[c][i][j + 1] + acc[c][i + 1][j] - acc[c][i][j] + flag[c][i][j];
            }
        }
    }

    int ans = 0;
    for (int i = 0; i < H; i++) {
        for (int j = 0; j < W; j++) {
            
            int tmax = min(H - i, W - j) + 1;
            int tmin = 0;
            while(tmax - tmin > 1) {
                int t = (tmax + tmin) / 2;
                int sum = 0;
                for(int c = 0; c < 26; c++) {
                    sum += (acc[c][i + t][j + t] - acc[c][i][j + t] - acc[c][i + t][j] + acc[c][i][j] > 0) ? 1 : 0;
                }
                if (sum <= K) {
                    tmin = t;
                } else {
                    tmax = t;
                }
            }
            int alpha = tmin;
            tmax = min(H - i, W - j) + 1;
            tmin = 0;
            while(tmax - tmin > 1) {
                int t = (tmax + tmin) / 2;
                int sum = 0;
                for(int c = 0; c < 26; c++) {
                    sum += (acc[c][i + t][j + t] - acc[c][i][j + t] - acc[c][i + t][j] + acc[c][i][j] > 0) ? 1 : 0;
                }
                if (sum < K) {
                    tmin = t;
                } else {
                    tmax = t;
                }
            }
            int beta = tmin;
            
            ans += alpha - beta;
        }
    }
    cout << ans << endl;

}

int main() {
    cin.tie(0);
    ios::sync_with_stdio(false);
    cout.setf(ios::fixed);
    cout.precision(16);
    solve();
    return 0;
}
0