結果

問題 No.1049 Zero (Exhaust)
ユーザー simkaren
提出日時 2020-05-27 11:53:40
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 99 ms / 2,000 ms
コード長 524 bytes
コンパイル時間 1,920 ms
コンパイル使用メモリ 181,660 KB
実行使用メモリ 57,856 KB
最終ジャッジ日時 2024-10-13 03:37:24
合計ジャッジ時間 4,108 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC optimize("O3", "unroll-loops")

#include <bits/stdc++.h>

using namespace std;

#define ll long long

constexpr ll mod = 1000000007LL;

int main(void){
    int p, k; cin >> p >> k;
    vector<vector<ll>> dp(k + 1, vector<ll>(2, 0LL));
    dp[0][0] = 1;
    for (int i = 1; i <= k; ++i){
        dp[i][0] = ((p - 1) * dp[i - 1][1] % mod * 2 + (p + 1) * dp[i - 1][0] % mod) % mod;
        dp[i][1] = (dp[i - 1][0] + 2 * (p - 1) % mod * dp[i - 1][1] % mod) % mod;
    }
    cout << dp[k][0] << endl;
    return 0;
}
0