結果

問題 No.1419 Power Moves
コンテスト
ユーザー SSRS
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff # 

#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;
    }
  }
}
0