結果

問題 No.866 レベルKの正方形
ユーザー ゆにぽけゆにぽけ
提出日時 2024-03-11 13:22:05
言語 C++23
(gcc 12.3.0 + boost 1.83.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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 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 3 ms
5,248 KB
testcase_05 AC 1 ms
5,248 KB
testcase_06 AC 1 ms
5,248 KB
testcase_07 AC 2 ms
5,248 KB
testcase_08 AC 3,156 ms
347,392 KB
testcase_09 AC 3,368 ms
347,264 KB
testcase_10 AC 4,779 ms
347,264 KB
testcase_11 AC 4,892 ms
347,264 KB
testcase_12 AC 5,378 ms
347,392 KB
testcase_13 AC 4,985 ms
347,264 KB
testcase_14 AC 5,388 ms
347,264 KB
testcase_15 AC 4,408 ms
347,264 KB
testcase_16 AC 5,445 ms
347,392 KB
testcase_17 AC 4,636 ms
347,264 KB
testcase_18 AC 4,840 ms
347,264 KB
testcase_19 AC 5,084 ms
347,264 KB
testcase_20 AC 4,277 ms
347,264 KB
testcase_21 AC 5,524 ms
347,264 KB
testcase_22 AC 1 ms
6,820 KB
testcase_23 AC 1 ms
6,816 KB
testcase_24 AC 2 ms
6,816 KB
権限があれば一括ダウンロードができます

ソースコード

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