結果

問題 No.1049 Zero (Exhaust)
ユーザー 沙耶花
提出日時 2020-05-08 21:34:26
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 121 ms / 2,000 ms
コード長 536 bytes
コンパイル時間 2,377 ms
コンパイル使用メモリ 196,580 KB
最終ジャッジ日時 2025-01-10 08:14:38
ジャッジサーバーID
(参考情報)
judge3 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
#define modulo 1000000007
#define mod(mod_x) ((((long long)mod_x+modulo))%modulo)
#define Inf 1000000

int main(){

	int P,K;
	cin>>P>>K;
	
	vector<vector<int>> dp(K+1,vector<int>(2,0));
	dp[0][0] = 1;
	
	for(int i=0;i<K;i++){
		dp[i+1][0] = mod(dp[i+1][0] + mod(dp[i][0]*(P+1)));
		dp[i+1][1] = mod(dp[i+1][1] + mod(dp[i][0]*(P-1)));
		dp[i+1][0] = mod(dp[i+1][0] + mod(dp[i][1]*2));
		dp[i+1][1] = mod(dp[i+1][1] + mod(dp[i][1]*mod(2*P-2)));
	}
	
	cout<<dp.back()[0]<<endl;
	
	return 0;
}
0