結果

問題 No.2994 べき内積
コンテスト
ユーザー 👑 ygussany
提出日時 2024-12-19 00:37:45
言語 C
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 2,391 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 492 ms
コンパイル使用メモリ 36,744 KB
実行使用メモリ 19,240 KB
最終ジャッジ日時 2024-12-19 00:37:57
合計ジャッジ時間 11,537 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 3 WA * 20
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#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;
	
	int j, k;
	long long B[1009][2001];
	for (i = 0; i <= N; i++) B[0][i] = A[i];
	for (k = 1; 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-1][i]);
		else for (i = 0; i <= N; i++) printf("0 ");
	} else {
		k = pow_mod(A[0], K[1]);
		for (i = 0; i <= N; i++) printf("%lld ", B[(K[0]+p-1)%p][i] * k % p);
	}
	printf("\n");
	fflush(stdout);
	return 0;
}
0