結果

問題 No.2983 Christmas Color Grid (Advent Calender ver.)
ユーザー 👑 ygussany
提出日時 2024-12-08 01:16:39
言語 C
(gcc 13.3.0)
結果
AC  
実行時間 594 ms / 3,340 ms
コード長 3,242 bytes
コンパイル時間 349 ms
コンパイル使用メモリ 33,888 KB
実行使用メモリ 18,176 KB
最終ジャッジ日時 2024-12-08 01:16:46
合計ジャッジ時間 4,991 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 64
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.c:5:11: warning: built-in function 'pow' declared as non-function [-Wbuiltin-declaration-mismatch]
    5 | long long pow[26];
      |           ^~~

ソースコード

diff #

#include <stdio.h>

const int bit[26] = {1, 2, 4, 8, 16, 32, 64, 128, 256, 512, 1024, 2048, 4096, 8192, 16384, 32768, 65536, 131072, 262144, 524288, 1048576, 2097152, 4194304, 8388608, 16777216, 33554432};
int Mod;
long long pow[26];

long long div_mod(long long x, long long y, long long z)
{
	if (x % y == 0) return x / y;
	else return (div_mod((1 + x / y) * y - x, (z % y), y) * z + x) / y;
}

long long pow_mod(int n, long long k)
{
	if (n == 0) return 0;
	
	long long N, ans = 1;
	for (N = n; k > 0; k >>= 1, N = N * N % Mod) if (k & 1) ans = ans * N % Mod;
	return ans;
}

long long calc(int H, int W, int x)
{
	int i, j, k, l = 0;
	static char S[30][30] = {};
	for (i = 1, k = 0; i <= H; i++) {
		for (j = 1; j <= W; j++, k++) {
			if ((x & bit[k]) != 0) {
				S[i][j] = '.';
				l++;
			} else S[i][j] = '#';
		}
	}
	
	int ii, jj;
	static int q[26][2], head, tail;
	long long ans = 0;
	for (ii = 1; ii <= H; ii++) {
		for (jj = 1; jj <= W; jj++) {
			if (S[ii][jj] == '#') continue;
			S[ii][jj] = '#';
			q[0][0] = ii;
			q[0][1] = jj;
			for (head = 0, tail = 1; head < tail; head++) {
				i = q[head][0];
				j = q[head][1];
				if (S[i-1][j] == '.') {
					S[i-1][j] = '#';
					q[tail][0] = i - 1;
					q[tail++][1] = j;
				}
				if (S[i+1][j] == '.') {
					S[i+1][j] = '#';
					q[tail][0] = i + 1;
					q[tail++][1] = j;
				}
				if (S[i][j-1] == '.') {
					S[i][j-1] = '#';
					q[tail][0] = i;
					q[tail++][1] = j - 1;
				}
				if (S[i][j+1] == '.') {
					S[i][j+1] = '#';
					q[tail][0] = i;
					q[tail++][1] = j + 1;
				}
			}
			ans += pow[tail];
		}
	}
	return ans % Mod;
}

char memo[33554432] = {};
long long ans = 0;

void DFS(int H, int W, int x, int l, char S[][30])
{
	if (memo[x] > 0) return;
	memo[x] = 1;
	
	int i, j, k = 0, head, tail = 0;
	int q[26][2];
	for (i = 1; i <= H; i++) {
		for (j = 1; j <= W; j++) {
			if (S[i][j] != '.') continue;
			if (S[i-1][j] == '#') {
				S[i-1][j] = '-';
				q[tail][0] = i - 1;
				q[tail++][1] = j;
			}
			if (S[i+1][j] == '#') {
				S[i+1][j] = '-';
				q[tail][0] = i + 1;
				q[tail++][1] = j;
			}
			if (S[i][j-1] == '#') {
				S[i][j-1] = '-';
				q[tail][0] = i;
				q[tail++][1] = j - 1;
			}
			if (S[i][j+1] == '#') {
				S[i][j+1] = '-';
				q[tail][0] = i;
				q[tail++][1] = j + 1;
			}
		}
	}
	ans += pow[l] * pow_mod(2, H * W - l - tail) % Mod;
	for (head = 0; head < tail; head++) {
		i = q[head][0];
		j = q[head][1];
		S[i][j] = '#';
	}
	for (head = 0; head < tail; head++) {
		i = q[head][0];
		j = q[head][1];
		S[i][j] = '.';
		DFS(H, W, x | bit[(i-1)*W+j-1], l + 1, S);
		S[i][j] = '#';
	}
}

int main()
{
	int H, W;
	long long K;
	scanf("%d %d %lld %d", &H, &W, &K, &Mod);
	
	int i, x;
	long long tmp, inv[26], invv = div_mod(1, pow_mod(2, H * W), Mod);
	for (i = 1, pow[0] = 0; i <= H * W; i++) {
		inv[i] = div_mod(1, i, Mod);
		pow[i] = pow_mod(i, K);
	}
	
	int j;
	char S[30][30] = {};
	for (i = 1; i <= H; i++) for (j = 1; j <= W; j++) S[i][j] = '#';
	for (i = 0; i < H * W; i++) {
		S[i/W+1][i%W+1] = '.';
		DFS(H, W, bit[i], 1, S);
		S[i/W+1][i%W+1] = '#';
	}
	for (i = 1, tmp = 0; i <= H * W; i++) tmp += inv[i];
	tmp %= Mod;
	ans = ans % Mod * invv % Mod * tmp % Mod;
	printf("%lld\n", ans % Mod);
	fflush(stdout);
	return 0;
}
0