結果
問題 | No.1049 Zero (Exhaust) |
ユーザー | yuji9511 |
提出日時 | 2020-05-08 22:11:19 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 15 ms / 2,000 ms |
コード長 | 830 bytes |
コンパイル時間 | 1,524 ms |
コンパイル使用メモリ | 167,004 KB |
実行使用メモリ | 19,020 KB |
最終ジャッジ日時 | 2024-07-04 00:47:58 |
合計ジャッジ時間 | 2,369 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 22 |
ソースコード
/*** author: yuji9511 ***/ #include <bits/stdc++.h> using namespace std; using ll = long long; using lpair = pair<ll, ll>; const ll MOD = 1e9+7; const ll INF = 1e18; #define rep(i,m,n) for(ll i=(m);i<(n);i++) #define rrep(i,m,n) for(ll i=(m);i>=(n);i--) #define printa(x,n) for(ll i=0;i<n;i++){cout<<(x[i])<<" \n"[i==n-1];}; void print() {} template <class H,class... T> void print(H&& h, T&&... t){cout<<h<<" \n"[sizeof...(t)==0];print(forward<T>(t)...);} ll dp[1000010][2] = {}; int main(){ cin.tie(0); ios::sync_with_stdio(false); ll P,K; cin >> P >> K; dp[0][0] = 1; rep(i,0,K){ dp[i+1][0] = dp[i][0] * (P+1) % MOD + dp[i][1] * 2 * (P-1); dp[i+1][0] %= MOD; dp[i+1][1] = dp[i][0] + (P-1) * 2 * dp[i][1]; dp[i+1][1] %= MOD; } ll ans = dp[K][0]; print(ans); }