結果

問題 No.2994 べき内積
ユーザー 👑 ygussanyygussany
提出日時 2024-12-19 01:14:52
言語 C
(gcc 12.3.0)
結果
WA  
実行時間 -
コード長 2,754 bytes
コンパイル時間 898 ms
コンパイル使用メモリ 38,792 KB
実行使用メモリ 19,248 KB
最終ジャッジ日時 2024-12-19 01:15:12
合計ジャッジ時間 17,508 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
16,504 KB
testcase_01 AC 3 ms
16,620 KB
testcase_02 AC 4 ms
16,432 KB
testcase_03 AC 4 ms
16,636 KB
testcase_04 AC 3 ms
18,300 KB
testcase_05 AC 3 ms
18,300 KB
testcase_06 AC 282 ms
18,180 KB
testcase_07 AC 464 ms
17,700 KB
testcase_08 AC 5 ms
16,552 KB
testcase_09 AC 415 ms
18,600 KB
testcase_10 AC 423 ms
17,928 KB
testcase_11 AC 303 ms
18,052 KB
testcase_12 AC 282 ms
18,692 KB
testcase_13 AC 444 ms
17,928 KB
testcase_14 AC 498 ms
17,928 KB
testcase_15 AC 300 ms
18,948 KB
testcase_16 AC 403 ms
18,824 KB
testcase_17 AC 292 ms
17,796 KB
testcase_18 AC 458 ms
17,624 KB
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 AC 1,796 ms
19,136 KB
testcase_24 WA -
testcase_25 AC 1,842 ms
19,112 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC target("avx2")
#pragma GCC optimize("Ofast,unroll-loops")
#include <stdio.h>

const int p = 1009;

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

#define MT_N 624
#define MT_M 397
#define MT_MATRIX_A 0x9908b0dfUL
#define MT_UPPER_MASK 0x80000000UL
#define MT_LOWER_MASK 0x7fffffffUL

static unsigned int mt[MT_N];
static int mti = MT_N + 1;

void init_genrand(unsigned int s)
{
    mt[0] = s & 0xffffffffUL;
    for (mti = 1; mti < MT_N; mti++) {
        mt[mti] = (1812433253UL * (mt[mti-1] ^ (mt[mti-1] >> 30)) + mti); 
        mt[mti] &= 0xffffffffUL;
    }
}

unsigned int genrand()
{
    unsigned int y;
    static unsigned int mag01[2] = {0x0UL, MT_MATRIX_A};

    if (mti >= MT_N) {
        int kk;
        if (mti == MT_N + 1) init_genrand(5489UL);
		
        for (kk = 0; kk < MT_N - MT_M; kk++) {
            y = (mt[kk] & MT_UPPER_MASK) | (mt[kk+1] & MT_LOWER_MASK);
            mt[kk] = mt[kk+MT_M] ^ (y >> 1) ^ mag01[y&0x1UL];
        }
        for (; kk < MT_N - 1; kk++) {
            y = (mt[kk] & MT_UPPER_MASK) | (mt[kk+1] & MT_LOWER_MASK);
            mt[kk] = mt[kk+(MT_M-MT_N)] ^ (y >> 1) ^ mag01[y&0x1UL];
        }
        y = (mt[MT_N-1] & MT_UPPER_MASK) | (mt[0] & MT_LOWER_MASK);
        mt[MT_N-1] = mt[MT_M-1] ^ (y >> 1) ^ mag01[y&0x1UL];

        mti = 0;
    }
  
    y = mt[mti++];

    y ^= (y >> 11);
    y ^= (y << 7) & 0x9d2c5680UL;
    y ^= (y << 15) & 0xefc60000UL;
    y ^= (y >> 18);

    return y;
}

int main()
{
	int i, M, N, K[400001], A[2001];
	scanf("%d %d", &M, &N);
	for (i = 0; i <= M; i++) scanf("%d", &(K[i]));
	for (i = 0; i <= N; i++) scanf("%d", &(A[i]));
	// for (i = 0, K[0] = p - 1; i <= N; i++) A[i] = genrand() % p;
	// A[0] = 2;
	
	int j, k;
	long long B[1009][2001];
	for (i = 1, B[0][0] = 1; i <= N; i++) B[0][i] = 0;
	for (i = 0; i <= N; i++) B[1][i] = A[i];
	for (k = 2; k < p; k++) {
		for (i = 0; i <= N; i++) for (j = 0, B[k][i] = 0; j <= i; j++) B[k][i] += B[k-1][j] * A[i-j];
		for (i = 0; i <= N; i++) B[k][i] %= p;
	}
	
	if (A[0] == 0) {
		for (i = M, k = 0; i >= 0 && k < p; i--) k = k * p + K[i];
		if (k < p) for (i = 0; i <= N; i++) printf("%lld ", B[k][i]);
		else for (i = 0; i <= N; i++) printf("0 ");
	} else {
		for (i = M, k = 0; i >= 1; i--) k = (k * p + K[i]) % (p - 1);
		k = pow_mod(A[0], k);
		for (i = 0; i <= N; i++) B[0][i] = 0;
		B[0][0] = k;
		B[0][p] = A[1] * k % p;
		for (k = 1; k <= K[0]; k++) {
			for (i = 0; i <= N; i++) for (j = 0, B[k][i] = 0; j <= i; j++) B[k][i] += B[k-1][j] * A[i-j];
			for (i = 0; i <= N; i++) B[k][i] %= p;
		}
		for (i = 0; i <= N; i++) printf("%lld ", B[K[0]][i]);
	}
	printf("\n");
	fflush(stdout);
	return 0;
}
0