結果

問題 No.866 レベルKの正方形
ユーザー ゆにぽけゆにぽけ
提出日時 2024-03-11 13:18:37
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 3,081 bytes
コンパイル時間 1,612 ms
コンパイル使用メモリ 137,012 KB
実行使用メモリ 352,384 KB
最終ジャッジ日時 2024-09-29 21:50:40
合計ジャッジ時間 14,327 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 2 ms
5,248 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 2 ms
5,248 KB
testcase_06 AC 2 ms
5,248 KB
testcase_07 AC 2 ms
5,248 KB
testcase_08 AC 2,063 ms
347,264 KB
testcase_09 AC 2,502 ms
347,264 KB
testcase_10 TLE -
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 <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];
					if(__builtin_popcount(cur) >= K)
					{
						ok = mid;
						continue;
					}
					cur |= dp[ni][j][a];
					if(__builtin_popcount(cur) >= K)
					{
						ok = mid;
						continue;
					}
					cur |= dp[i][nj][a];
					if(__builtin_popcount(cur) >= K)
					{
						ok = mid;
						continue;
					}
					cur |= dp[ni][nj][a];
					if(__builtin_popcount(cur) >= K)
					{
						ok = mid;
						continue;
					}
					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];
					if(__builtin_popcount(cur) > K)
					{
						ok = mid;
						continue;
					}
					cur |= dp[ni][j][a];
					if(__builtin_popcount(cur) > K)
					{
						ok = mid;
						continue;
					}
					cur |= dp[i][nj][a];
					if(__builtin_popcount(cur) > K)
					{
						ok = mid;
						continue;
					}
					cur |= dp[ni][nj][a];
					if(__builtin_popcount(cur) > K)
					{
						ok = mid;
						continue;
					}
					ng = mid;
					/* 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