結果
問題 | No.1049 Zero (Exhaust) |
ユーザー |
|
提出日時 | 2020-05-08 21:35:32 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 34 ms / 2,000 ms |
コード長 | 553 bytes |
コンパイル時間 | 2,184 ms |
コンパイル使用メモリ | 191,712 KB |
最終ジャッジ日時 | 2025-01-10 08:16:40 |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 22 |
ソースコード
#include <bits/stdc++.h> using namespace std; const int64_t MOD = 1e9+7; void add(int64_t& a, int64_t b){ a = (a+b) % MOD; } void mul(int64_t& a, int64_t b){ a = a*b % MOD; } int main(){ int64_t P, K; cin >> P >> K; static int64_t dp[1000001][2]; dp[0][0] = 1; for(int i=0; i<K; i++){ add(dp[i+1][0], dp[i][0]+dp[i][1]); add(dp[i+1][1], (P-1)*(dp[i][0]+dp[i][1])); add(dp[i+1][0], P*dp[i][0] + dp[i][1]); add(dp[i+1][1], (P-1)*dp[i][1]); } cout << dp[K][0] << endl; return 0; }