結果

問題 No.866 レベルKの正方形
コンテスト
ユーザー ok
提出日時 2020-07-11 09:19:28
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 3,618 ms / 6,000 ms
コード長 1,667 bytes
コンパイル時間 769 ms
コンパイル使用メモリ 82,572 KB
実行使用メモリ 415,044 KB
最終ジャッジ日時 2024-10-12 15:46:02
合計ジャッジ時間 50,302 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<string>
#include<iomanip>
#include<cmath>
#include<vector>
#include<algorithm>

using namespace std;

// #define int short
#define endl "\n"

const long long INF = (long long)1e18;
const long long MOD = (long long)1e9 + 7; 

string yn(bool f){return f?"Yes":"No";}
string YN(bool f){return f?"YES":"NO";}

const int NUM = 26;
int H, W, K;
int con[2005][2005][26];
string squares[2000];

inline int check(int y, int x, int sz){
	sz--;
	
	if(sz < 0) return 0;
	
	int y2 = y + sz, x2 = x + sz;
	
	if(y2 >= H || x2 >= W) return NUM+1;
	
	int level = 0;
	
	for(int i = 0; i < NUM; i++){
		level += !!(con[y2+1][x2+1][i] - con[y2+1][x][i] - con[y][x2+1][i] + con[y][x][i]);
	}
	
	return level;
}

signed main(){
	cin.tie(0);
	ios::sync_with_stdio(false);
	// cout<<fixed<<setprecision(10);
	
	long long ans = 0;
	
	
	cin>>H>>W>>K;
	
	for(int i = 0; i < H; i++){
		cin>>squares[i];
		for(int j = 0; j < W; j++){
			con[i+1][j+1][squares[i][j] - 'a']++;
		}
	}
	
	for(int k = 0; k < NUM; k++){
		for(int i = 0; i <= H; i++){
			for(int j = 0; j <= W; j++){
				con[i][j+1][k] += con[i][j][k];
				if(i) con[i][j][k] += con[i-1][j][k];
			}
		}
	}
	
	for(int i = 0; i < H; i++){
		for(int j = 0; j < W; j++){
			int minimum = (H-i < W-j)?H-i:W-j;
			
			int h = minimum + 1, l = 0, m, a;
			
			while(l + 1 < h){
				m = (l + h)/2;
				
				if(check(i,j,m) > K){
					h = m;
				} else {
					l = m;
				}
			}
			
			a = h;
			
			h = minimum + 1, l = 0;
			
			while(l + 1 < h){
				m = (l + h)/2;
				
				if(!(check(i,j,m) < K)){
					h = m;
				} else {
					l = m;
				}
			}
			
			ans += a - l - 1;
		}
	}
	
	cout<<ans<<endl;
	
	return 0;
}
0