結果

問題 No.866 レベルKの正方形
コンテスト
ユーザー ok
提出日時 2019-08-18 20:08:59
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
TLE  
実行時間 -
コード長 2,197 bytes
コンパイル時間 1,047 ms
コンパイル使用メモリ 89,420 KB
実行使用メモリ 351,872 KB
最終ジャッジ日時 2024-10-01 14:14:17
合計ジャッジ時間 16,043 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 3
other AC * 8 TLE * 2 -- * 12
権限があれば一括ダウンロードができます

ソースコード

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 check(vector<vector<vector<short>>> & con, int y, int x, int sz){
	int level = 0;
	
	sz--;
	
	if(sz < 0) return 0;
	if(sz + y >= H || sz + x >= W) return NUM+1;
	
	int y2 = y + sz, x2 = x + sz;
	
	for(int i = 0; i < NUM; i++){
		int sum = con[y2+1][x2+1][i] - con[y2+1][x][i] - con[y][x2+1][i] + con[y][x][i];
		
		level += !!sum;
	}
	
	return level;
}

signed main(){
	cin.tie(0);
	ios::sync_with_stdio(false);
	cout<<fixed<<setprecision(10);
	
	long long ans = 0;
	
	vector<string> squares;
	vector<vector<vector<short>>> con;
	
	cin>>H>>W>>K;
	
	
	con.resize(H+2, vector<vector<short>>(W+2, vector<short>(NUM, 0)));
	squares.resize(H);
	
	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];
				// con[i+1][j][k] += con[i][j][k];
				if(i) con[i][j][k] += con[i-1][j][k];
			}
		}
	}
	// for(int k = 0; k < NUM; k++){
		// for(int i = 0; i <= H; i++){
			// for(int j = 0; j <= W; j++){
				// con[i+1][j][k] += con[i][j][k];
			// }
		// }
	// }
	
	// while(true){
		// int a, b, c;
		// cin>>a>>b>>c;
		
		// cout<<check(con, a, b, c)<<endl;
	// }
	
	for(int i = 0; i < H; i++){
		for(int j = 0; j < W; j++){
			int h = min(H-i, W-j) + 1, l = 0, m;
			int a, b;
			
			while(l + 1 < h){
				m = (l + h)/2;
				
				if(check(con,i,j,m) > K){
					h = m;
				} else {
					l = m;
				}
			}
			
			a = h;
			
			h = min(H-i, W-j) + 1, l = 0;
			
			while(l + 1 < h){
				m = (l + h)/2;
				
				if(!(check(con,i,j,m) < K)){
					h = m;
				} else {
					l = m;
				}
			}
			
			b = l;
			
			// cout<<"i = "<<i<<" j = "<<j<<" a = "<<a<<" b = "<<b<<endl;
			ans += a - b - 1;
		}
	}
	
	cout<<ans<<endl;
	
	return 0;
}
0