#478483 (C++14) No.1049 Zero (Exhaust)

• 問題: No.1049 Zero (Exhaust)
• ユーザー: auaua
• 提出日時: 2020-05-08 21:51:07
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 91 ms / 2,000 ms
• コード長: 801 bytes
• コンパイル時間: 1,373 ms
• コンパイル使用メモリ: 170,660 KB
• 実行使用メモリ: 57,976 KB
• 最終ジャッジ日時: 2024-07-04 00:30:45
• 合計ジャッジ時間: 3,303 ms
• ジャッジサーバーID （参考情報）: judge3 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 6,812 KB
testcase_01	AC	1 ms; 6,940 KB
testcase_02	AC	1 ms; 6,940 KB
testcase_03	AC	59 ms; 40,484 KB
testcase_04	AC	1 ms; 5,376 KB
testcase_05	AC	50 ms; 33,580 KB
testcase_06	AC	91 ms; 57,976 KB
testcase_07	AC	13 ms; 10,112 KB
testcase_08	AC	25 ms; 19,400 KB
testcase_09	AC	69 ms; 44,108 KB
testcase_10	AC	72 ms; 47,880 KB
testcase_11	AC	71 ms; 48,000 KB
testcase_12	AC	85 ms; 55,040 KB
testcase_13	AC	20 ms; 14,840 KB
testcase_14	AC	37 ms; 26,464 KB
testcase_15	AC	63 ms; 41,344 KB
testcase_16	AC	51 ms; 33,920 KB
testcase_17	AC	54 ms; 35,456 KB
testcase_18	AC	75 ms; 48,752 KB
testcase_19	AC	64 ms; 41,600 KB
testcase_20	AC	27 ms; 18,944 KB
testcase_21	AC	65 ms; 42,752 KB
testcase_22	AC	38 ms; 25,464 KB
testcase_23	AC	91 ms; 57,856 KB
testcase_24	AC	88 ms; 57,940 KB

ソースコード

#include<bits/stdc++.h>
#include<unordered_set>
#include<unordered_map>
using namespace std;
#define int long long
#define REP(i,m,n) for(int i=(m);i<(n);i++)
#define rep(i,n) REP(i,0,n)
#define pb push_back
#define all(a) a.begin(),a.end()
#define rall(c) (c).rbegin(),(c).rend()
#define mp make_pair
#define endl '\n'
typedef long long ll;
typedef pair<ll,ll> pll;
typedef long double ld;
const ll inf=1e9+7;
const ll mod=1e9+7;
signed main(){
    ll p,k;cin>>p>>k;
    vector<vector<ll> >dp(k+1,vector<ll>(2));
    dp[0][0]=1;
    rep(i,k){
        dp[i+1][1]=(dp[i+1][1]+dp[i][0]*(p-1)%mod)%mod;
        dp[i+1][0]=(dp[i+1][0]+dp[i][0]*(p+1)%mod)%mod;
        dp[i+1][0]=(dp[i+1][0]+dp[i][1]*(2)%mod)%mod;
        dp[i+1][1]=(dp[i+1][1]+dp[i][1]*(2*p-2)%mod)%mod;
    }
    cout<<dp[k][0]<<endl;
}