結果

問題 No.866 レベルKの正方形
ユーザー 👑 emthrmemthrm
提出日時 2019-08-16 22:34:59
言語 C++17
(gcc 13.2.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 2,455 bytes
コンパイル時間 2,231 ms
コンパイル使用メモリ 128,272 KB
実行使用メモリ 418,048 KB
最終ジャッジ日時 2023-10-24 01:49:38
合計ジャッジ時間 10,368 ms
ジャッジサーバーID
(参考情報)
judge15 / judge11
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
6,692 KB
testcase_01 AC 3 ms
6,692 KB
testcase_02 AC 3 ms
6,692 KB
testcase_03 AC 4 ms
6,692 KB
testcase_04 AC 3 ms
6,692 KB
testcase_05 AC 4 ms
6,692 KB
testcase_06 AC 3 ms
6,692 KB
testcase_07 AC 4 ms
6,692 KB
testcase_08 TLE -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
権限があれば一括ダウンロードができます

ソースコード

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 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 (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;
  }
  return res;
}

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

  int h, w, k; 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];
  }
  // REP(i, h) {
  //   REP(j, w) cout << cnt[1][i][j] << ' ';
  //   cout << '\n';
  // }
  long long ans = 0;
  REP(i, h) REP(j, w) {
    int tmp = (i < j ? h - (j - i) - 1 : h - 1);
    // cout << tmp << (j == w - 1 ? '\n' : ' ');
    if (level(i, j, tmp, j + tmp - i) < k) continue;
    int lb = i - 1, ub = tmp;
    while (ub - lb > 1) {
      int mid = (lb + ub) / 2, l = level(i, j, mid, j + mid - i);
      (l >= k ? ub : lb) = mid;
    }
    int mn = ub;
    lb = i - 1; ub = tmp;
    while (ub - lb > 1) {
      int mid = (lb + ub) / 2, l = level(i, j, mid, j + mid - i);
      (l > k ? ub : lb) = mid;
    }
    int mx = ub + (level(i, j, tmp, j + tmp - i) <= k ? 1 : 0);
    ans += mx - mn;
    // cout << i << ' ' << j << ' ' << mn << ' ' << mx << '\n';
  }
  cout << ans << '\n';
  return 0;
}
0