結果

問題 No.866 レベルKの正方形
コンテスト
ユーザー emthrm
提出日時 2019-08-16 23:06:31
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 2,755 bytes
コンパイル時間 1,353 ms
コンパイル使用メモリ 126,316 KB
最終ジャッジ日時 2025-01-07 12:48:00
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 11 TLE * 11
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <bitset>
#include <cassert>
#include <cctype>
#include <chrono>
#define _USE_MATH_DEFINES
#include <cmath>
#include <cstdio>
#include <cstring>
#include <ctime>
#include <deque>
#include <functional>
#include <iostream>
#include <iterator>
#include <map>
#include <queue>
#include <random>
#include <set>
#include <sstream>
#include <string>
#include <tuple>
#include <utility>
#include <vector>
using namespace std;

#define FOR(i,m,n) for(int i=(m);i<(n);++i)
#define REP(i,n) FOR(i,0,n)
#define ALL(v) (v).begin(),(v).end()

const int INF = 0x3f3f3f3f;
const long long LINF = 0x3f3f3f3f3f3f3f3fLL;
const double EPS = 1e-8;
const int MOD = 1000000007; // 998244353;
const int dy[] = {1, 0, -1, 0}, dx[] = {0, -1, 0, 1};
/*-------------------------------------------------*/
int k, cnt[26][2000][2000] = {};
int level(int y1, int x1, int y2, int x2) {
  int res = 0;
  REP(i, 26) {
    int tmp = cnt[i][y2][x2];
    if (tmp == 0) continue;
    if (y1 > 0) tmp -= cnt[i][y1 - 1][x2];
    if (x1 > 0) tmp -= cnt[i][y2][x1 - 1];
    if (y1 > 0 && x1 > 0) tmp += cnt[i][y1 - 1][x1 - 1];
    if (tmp > 0) ++res;
    if (res + (25 - i) < k) break;
    if (res > k) break;
  }
  return res;
}

int main() {
  cin.tie(0); ios::sync_with_stdio(false);
  // freopen("input.txt", "r", stdin);

  int h, w; cin >> h >> w >> k;
  vector<vector<char> > c(h, vector<char>(w)); REP(i, h) REP(j, w) cin >> c[i][j];
  REP(i, h) {
    REP(j, w) {
      REP(k, 26) if (j > 0) cnt[k][i][j] = cnt[k][i][j - 1];
      ++cnt[c[i][j] - 'a'][i][j];
    }
  }
  REP(j, w) {
    FOR(i, 1, h) REP(k, 26) cnt[k][i][j] += cnt[k][i - 1][j];
  }
  long long ans = 0;
  REP(i, h) REP(j, w) {
    if ((h - i) * (w - j) < k) continue;
    int tmp = (i < j ? h - (j - i) - 1 : h - 1);
    vector<int> memo(tmp - i + 1, -1);
    memo[tmp - i] = level(i, j, tmp, j + tmp - i);
    // cout << tmp << (j == w - 1 ? '\n' : ' ');
    if (memo[tmp - i] < k) continue;
    int lb = i - 1, ub = tmp;
    while (ub - lb > 1) {
      int mid = (lb + ub) / 2;
      int l = (memo[mid - i] == -1 ? level(i, j, mid, j + mid - i) : memo[mid - i]);
      memo[mid - i] = l;
      (l >= k ? ub : lb) = mid;
    }
    int mn = ub;
    lb = i - 1; ub = tmp;
    if (memo[tmp - i] > k) {
      while (ub - lb > 1) {
        int mid = (lb + ub) / 2;
        int l = (memo[mid - i] == -1 ? level(i, j, mid, j + mid - i) : memo[mid - i]);
        memo[mid - i] = l;
        (l > k ? ub : lb) = mid;
      }
    }
    int mx = ub + (memo[tmp - i] <= k ? 1 : 0);
    ans += mx - mn;
    // cout << i << ' ' << j << ' ' << mn << ' ' << mx << '\n';
  }
  cout << ans << '\n';
  return 0;
}
0