結果
問題 | No.1419 Power Moves |
ユーザー |
![]() |
提出日時 | 2021-03-05 21:51:43 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 129 ms / 2,000 ms |
コード長 | 1,567 bytes |
コンパイル時間 | 1,971 ms |
コンパイル使用メモリ | 170,012 KB |
実行使用メモリ | 5,248 KB |
最終ジャッジ日時 | 2024-10-07 01:31:54 |
合計ジャッジ時間 | 5,459 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 31 |
ソースコード
#include <bits/stdc++.h>using namespace std;const long long MOD = 1000000007;long long modpow(long long a, long long b, long long mod){long long ans = 1;while (b > 0){if (b % 2 == 1){ans *= a;ans %= mod;}a *= a;a %= mod;b /= 2;}return ans;}long long modinv(long long a){return modpow(a, MOD - 2, MOD);}int main(){int N, K;cin >> N >> K;if (N % 2 == 1){vector<long long> cnt(N, 0);long long r = modpow(2, K - 1, N);long long q = (modpow(2, K - 1, MOD) - r + MOD) % MOD * modinv(N) % MOD;for (int i = 0; i < N; i++){cnt[i] += q * 2;cnt[i] %= MOD;}for (int i = 0; i < r; i++){int p = (i * 2 + 1) % N;cnt[p]++;cnt[p] %= MOD;cnt[(N - p) % N]++;cnt[(N - p) % N] %= MOD;}long long C = modinv(modpow(2, K, MOD));for (int i = 0; i < N; i++){cnt[i] *= C;cnt[i] %= MOD;cout << cnt[i] << endl;}} else {vector<long long> cnt(N, 0);long long r = modpow(2, K - 1, N / 2);long long q = (modpow(2, K - 1, MOD) - r + MOD) % MOD * modinv(N / 2) % MOD;for (int i = 0; i < N / 2; i++){cnt[i * 2 + 1] += q * 2;cnt[i * 2 + 1] %= MOD;}for (int i = 0; i < r; i++){int p = (i * 2 + 1) % N;cnt[p]++;cnt[p] %= MOD;cnt[(N - p) % N]++;cnt[(N - p) % N] %= MOD;}long long C = modinv(modpow(2, K, MOD));for (int i = 0; i < N; i++){cnt[i] *= C;cnt[i] %= MOD;cout << cnt[i] << endl;}}}