結果

問題 No.1049 Zero (Exhaust)
コンテスト
ユーザー hipotaf
提出日時 2020-06-06 18:50:54
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 35 ms / 2,000 ms
コード長 1,448 bytes
コンパイル時間 585 ms
コンパイル使用メモリ 67,456 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-12-23 13:32:29
合計ジャッジ時間 2,106 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <string>
#include <vector>

using namespace std;

using ll = long long;

template <class T>
using grid = vector<vector<T>>;

#define REP(i,n) for(ll i=0;i<(ll)(n);i++)
#define REPD(i,n) for(ll i=n-1;i>=0;i--)
#define FOR(i,a,b) for(ll i=a;i<=(ll)(b);i++)
#define FORD(i,a,b) for(ll i=a;i>=(ll)(b);i--)

#define input(...) __VA_ARGS__; in(__VA_ARGS__)

void print() {
  std::cout << std::endl;
}

template <class Head, class... Tail>
void print(Head&& head, Tail&&... tail) {
  std::cout << head << " ";
  print(std::forward<Tail>(tail)...);
}

void in() { }

template <class Head, class... Tail>
void in(Head&& head, Tail&&... tail) {
  cin >> head;
  in(std::forward<Tail>(tail)...);
}

int P;
int f(int n, int k) {
  if (k == 0) return n == 0;
  int ans = 0;
  REP(i, P) {

    int v1 = f((n + i) % P, k - 1);
    int v2 = f((n * i) % P, k - 1);
    print(n, k, i, "+:", v1);
    print(n, k, i, "*:", v2);
    ans += v1 + v2;
  }
  return ans;
}

ll MOD = 1'000'000'000 + 7;
ll mod(ll v) {
  if (v >= MOD) return v % MOD;
  return v % MOD;
}

int main() {
  int input(p, k);
  P = p;

  ll zero_pattern = 1;
  ll other_pattern = 0;
  ll to_zero = 0;
  ll to_other = 0;
  REP(i, k) {
    to_zero = mod(zero_pattern * (p + 1) + other_pattern * 2);
    to_other = mod(zero_pattern * (p - 1)) + mod(other_pattern * (p - 1 + p - 1));

    zero_pattern = to_zero;
    other_pattern = to_other;
  }
  print(zero_pattern);
}
0