結果

問題 No.1956 猫の額
ユーザー 37zigen37zigen
提出日時 2022-05-22 21:23:27
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 3,083 bytes
コンパイル時間 1,221 ms
コンパイル使用メモリ 92,732 KB
実行使用メモリ 6,088 KB
最終ジャッジ日時 2023-10-20 18:18:07
合計ジャッジ時間 52,742 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 4,295 ms
6,088 KB
testcase_01 AC 465 ms
6,084 KB
testcase_02 AC 4,339 ms
6,088 KB
testcase_03 AC 489 ms
6,088 KB
testcase_04 AC 6,098 ms
6,088 KB
testcase_05 AC 4,807 ms
6,088 KB
testcase_06 AC 6,203 ms
6,088 KB
testcase_07 AC 2,583 ms
6,088 KB
testcase_08 WA -
testcase_09 AC 484 ms
5,776 KB
testcase_10 WA -
testcase_11 AC 677 ms
4,716 KB
testcase_12 AC 1,287 ms
4,976 KB
testcase_13 AC 96 ms
4,348 KB
testcase_14 AC 270 ms
4,348 KB
testcase_15 AC 449 ms
5,976 KB
testcase_16 AC 4,070 ms
5,848 KB
testcase_17 WA -
testcase_18 WA -
testcase_19 AC 2,775 ms
6,088 KB
testcase_20 AC 2,774 ms
6,088 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function 'int primitive_root(int)':
main.cpp:48:1: warning: control reaches end of non-void function [-Wreturn-type]
   48 | }
      | ^

ソースコード

diff # 

#include <algorithm>
#include <cassert>
#include <limits>
#include <queue>
#include <vector>
#include <iostream>
#include <assert.h>

using namespace std;
using i64 = long long;

const int NMAX = 100;
const int AMAX = 100000;
vector<int> A;
int N, M, C, sumA;

i64 powmod(i64 a, i64 n, i64 p) {
	if (n == 0) return 1;
	return powmod(a * a % p, n / 2, p) * (n % 2 == 1 ? a : 1) % p;
}

i64 inv(i64 a, i64 p) {
	return powmod(a, p - 2, p);
}

bool is_prime(int a) {
	for (int div = 2; div * div <= a; ++div) if (a % div == 0) return false;
	return true;
}

int primitive_root(int p) {
	vector<int> div;
	for (int i = 2; i * i <= p - 1; ++i) {
		if ((p - 1) % i == 0) {
			div.push_back(i);
			div.push_back((p - 1) / i);
		}
	}

	for (int i = 1; i < p - 1; ++i) {
		bool ok = true;
		for (int d : div) {
			ok &= powmod(i, d, p) != 1;
		}
		if (ok) return i;
	}

}

int garner(vector<int>& a, vector<int>& mods, int m) {
	int n = a.size();
	vector<i64> b;
	auto gen_base = [&mods](int mod) {
		vector<i64> base{ 1 };
		for (int j = 0; j < mods.size(); ++j) {
			base.push_back(base.back() * mods[j] % mod);
		}
		return base;
	};
	auto f = [&b](vector<i64> base, int mod) {
		i64 x = 0;
		for (int j = 0; j < b.size(); ++j) {
			x += b[j] * base[j];
		}
		x %= mod;
		return x;
	};
	for (int i = 0; i < n; ++i) {
		auto base = gen_base(mods[i]);
		b.push_back(inv(base[i], mods[i]) * (-f(base, mods[i]) + a[i]) % mods[i]);
		if (b[i] < 0) b[i] += mods[i];
	}
	return (int) f(gen_base(m), m);
}

void solve(int mod, vector<int>& ans) {
	int g = primitive_root(mod);
	int r = powmod(g, (mod - 1) / C, mod);
	int zeta = 1;
	for (int i = 0; i < C; ++i, zeta = zeta * r % mod) {
		vector<int> f(ans.size());
		f[0] = 1;
		int upper = 0;
		for (int a : A) {
			for (int j = upper; j >= 0; --j) if (f[j] != 0) {
				f[j + a] += zeta * f[j];
				f[j + a] %= mod;
			}
			upper += a;
		}
		for (int j = 0; j < ans.size(); ++j) ans[j] += f[j];
	}
	for (int i = 0; i < ans.size(); ++i) ans[i] %= mod;
	ans[0] -= C;
	int invC = inv(C, mod);
	for (int i = 0; i < ans.size(); ++i) ans[i] = invC * ans[i] % mod;
}


int main() {

	cin >> N >> M >> C;
	assert(1 <= N && N <= 90);
	assert(2 <= M && M <= 1e9);
	assert(1 <= C && C <= N);
	
	A.resize(N);
	for (int i = 0; i < N; ++i) cin >> A[i];
	for (int a : A) sumA += a;
	for (int a : A) assert(1 <= a);
	assert(1 <= sumA && sumA <= 1e5);
	bool reversed = 2 * C < N;
	if (reversed) C = N - C;

	vector<int> mods;
	int num = 6;
	for (int m = 14000 / C * C + 1; mods.size() != num; m += C) {
		if (is_prime(m)) {
			mods.push_back(m);
		}
	}
	vector<vector<int>> x(num, vector<int>(sumA + 1));
	for (int i = 0; i < mods.size(); ++i) {
		solve(mods[i], x[i]);
	}
	vector<int> ans(sumA + 1);
	for (int i = 0; i <= sumA; ++i) {
		vector<int> a;
		for (int j = 0; j < mods.size(); ++j) a.push_back(x[j][i]);
		ans[i] = garner(a, mods, M);
	}
	if (reversed) {
		for (int i = 0; i <= sumA / 2; ++i) {
			swap(ans[i], ans[sumA - i]);
		}
	}

	if (N == 2 * C) ans[sumA] -= 1;

	for (int i = 1; i <= sumA; ++i) printf("%d%c", ans[i], i == sumA ? '\n' : ' ');

}
0