結果

問題 No.866 レベルKの正方形
コンテスト
ユーザー ゆにぽけ
提出日時 2024-03-11 13:22:05
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 5,524 ms / 6,000 ms
コード長 2,315 bytes
コンパイル時間 1,667 ms
コンパイル使用メモリ 157,096 KB
実行使用メモリ 347,392 KB
最終ジャッジ日時 2024-09-29 21:51:50
合計ジャッジ時間 18,614 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
#include <array>
#include <iterator>
#include <string>
#include <cctype>
#include <cstring>
#include <cstdlib>
#include <cassert>
#include <cmath>
#include <ctime>
#include <iomanip>
#include <numeric>
#include <stack>
#include <queue>
#include <map>
#include <unordered_map>
#include <set>
#include <unordered_set>
#include <bitset>
#include <random>
#include <utility>
#include <functional>
using namespace std;
void Main()
{
	int H,W,K;
	cin >> H >> W >> K;
	int L = 1;
	while((1 << L) < min(H,W))
	{
		L++;
	}
	vector<vector<vector<int>>> dp(H,vector<vector<int>>(W,vector<int>(L)));
	for(int i = 0;i < H;i++)
	{
		string S;
		cin >> S;
		for(int j = 0;j < W;j++)
		{
			dp[i][j][0] = 1 << (S[j] - 'a');
		}
	}
	for(int k = 1;k < L;k++)
	{
		int t = 1 << (k - 1);
		for(int i = 0;i + 2 * t <= H;i++)
		{
			for(int j = 0;j + 2 * t <= W;j++)
			{
				dp[i][j][k] = dp[i][j][k - 1] | dp[i + t][j][k - 1] | dp[i][j + t][k - 1] | dp[i + t][j + t][k - 1];
			}
		}
	}
	vector<int> A(min(H,W) + 1);
	for(int i = 1;i < (int)A.size();i++)
	{
		while(1 << (A[i] + 1) <= i)
		{
			A[i]++;
		}
	}
	long long ans = 0;
	for(int i = 0;i < H;i++)
	{
		for(int j = 0;j < W;j++)
		{
			int lo = 0,up = 0;
			{
				int ng = 0,ok = min(H - i,W - j) + 1;
				while(ok - ng > 1)
				{
					int mid = (ok + ng) / 2;
					int a = A[mid];
					int cur = 0;
					int ni = i + mid - (1 << a);
					int nj = j + mid - (1 << a);
					cur |= dp[i][j][a];
					cur |= dp[ni][j][a];
					cur |= dp[i][nj][a];
					cur |= dp[ni][nj][a];
					if(__builtin_popcount(cur) >= K)
					{
						ok = mid;
					}
					else
					{
						ng = mid;
					}
				}
				lo = ok;
			}
			{
				int ng = 0,ok = min(H - i,W - j) + 1;
				while(ok - ng > 1)
				{
					int mid = (ok + ng) / 2;
					int a = A[mid];
					int cur = 0;
					int ni = i + mid - (1 << a);
					int nj = j + mid - (1 << a);
					cur |= dp[i][j][a];
					cur |= dp[ni][j][a];
					cur |= dp[i][nj][a];
					cur |= dp[ni][nj][a];
					if(__builtin_popcount(cur) > K)
					{
						ok = mid;
					}
					else
					{
						ng = mid;
					}
				}
				up = ok;
			}
			ans += up - lo;
		}
	}
	cout << ans << "\n";
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int tt = 1;
	/* cin >> tt; */
	while(tt--) Main();
}
0