結果

問題 No.1049 Zero (Exhaust)
コンテスト
ユーザー kenta255
提出日時 2020-05-08 23:36:10
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 30 ms / 2,000 ms
コード長 1,264 bytes
コンパイル時間 2,207 ms
コンパイル使用メモリ 194,484 KB
最終ジャッジ日時 2025-01-10 09:23:44
ジャッジサーバーID
（参考情報） 		judge2 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
//#define P pair<ll,ll>
#define FOR(I,A,B) for(ll I = ll(A); I < ll(B); ++I)
#define FORR(I,A,B) for(ll I = ll((B)-1); I >= ll(A); --I)
#define TO(x,t,f) ((x)?(t):(f))
#define SORT(x) (sort(x.begin(),x.end())) // 0 2 2 3 4 5 8 9
#define POSL(x,v) (lower_bound(x.begin(),x.end(),v)-x.begin()) //xi>=v  x is sorted
#define POSU(x,v) (upper_bound(x.begin(),x.end(),v)-x.begin()) //xi>v  x is sorted
#define NUM(x,v) (POSU(x,v)-POSL(x,v))  //x is sorted
#define REV(x) (reverse(x.begin(),x.end())) //reverse
ll gcd(ll a,ll b){if(a%b==0)return b;return gcd(b,a%b);}
ll lcm(ll a,ll b){ll c=gcd(a,b);return ((a/c)*(b/c)*c);}
#define NEXTP(x) next_permutation(x.begin(),x.end())
const ll INF=ll(1e16)+ll(7);
const ll MOD=1000000007LL;
#define out(a) cout<<fixed<<setprecision((a))
//tie(a,b,c) = make_tuple(10,9,87);
#define pop(a) __builtin_popcount((a))
ll keta(ll a){ll r=0;while(a){a/=10;r++;}return r;}


int main(){


  ll P,K;
  cin >> P >> K;

  vector<ll> dp0(K+1,0),dp1(K+1,0);

  dp0[0] = 1;
  dp1[0] = 0;

  FOR(i,1,K+1){
    dp0[i] = (dp0[i-1]*(P+1)%MOD + dp1[i-1]*(1+1)%MOD   ) % MOD;
    dp1[i] = (dp0[i-1]*(0+P-1)%MOD + dp1[i-1]*(P-1+P-1)%MOD   ) % MOD;
  }

  cout << dp0[K] << endl;

}
0