結果

問題 No.866 レベルKの正方形
ユーザー mkawa2mkawa2
提出日時 2021-02-27 16:33:40
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,557 bytes
コンパイル時間 1,762 ms
コンパイル使用メモリ 195,128 KB
最終ジャッジ日時 2025-01-19 08:01:40
ジャッジサーバーID
（参考情報） 		judge2 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 8 TLE * 14
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define rep(i, n) for (int i = 0; i < (n); i++)
#define all(x) (x).begin(), (x).end()
using ll = long long;
using pii = pair<int, int>;
using vi = vector<int>;
using vll = vector<ll>;
using vvi = vector<vector<int>>;
const int inf = 1e9;

int cs[26][2005][2005];

int main() {
  cin.tie(0);
  ios::sync_with_stdio(false);

  int h, w, k;
  cin >> h >> w >> k;
  rep(i, h) rep(j, w) {
    char c;
    cin >> c;
    cs[c - 'a'][i + 1][j + 1] = 1;
  }
  rep(a, 26) {
    rep(i, h) rep(j, w) cs[a][i + 1][j + 1] += cs[a][i + 1][j];
    rep(j, w) rep(i, h) cs[a][i + 1][j + 1] += cs[a][i][j + 1];
  }

  auto ok1=[&](int const i, int const j, int const m) {
    int cnt = 0;
    rep(a, 26) if (cs[a][i][j] + cs[a][i + m][j + m] - cs[a][i + m][j] -
                   cs[a][i][j + m]) cnt++;
    return cnt < k;
  };
  auto ok2=[&](int const i, int const j, int const m) {
    int cnt = 0;
    rep(a, 26) if (cs[a][i][j] + cs[a][i + m][j + m] - cs[a][i + m][j] -
                   cs[a][i][j + m]) cnt++;
    return cnt <= k;
  };

  int ans = 0;
  rep(i, h) rep(j, w) {
    int l = 0, r = min(h - i, w - j) + 1;
    while (l + 1 < r) {
      int m = (l + r) / 2;
      if (ok1(i, j, m))
        l = m;
      else
        r = m;
    }
    int L=l;

    r = min(h - i, w - j) + 1;
    while (l + 1 < r) {
      int m = (l + r) / 2;
      if (ok2(i, j, m))
        l = m;
      else
        r = m;
    }
    int R=l;
    ans+=R-L;
    //cout<<i<<" "<<j<<" "<<L<<" "<<R<<" "<<endl;
  }
  cout<<ans<<endl;
  return 0;
}
0