結果

問題 No.1049 Zero (Exhaust)
ユーザー mamimume____momamimume____mo
提出日時 2020-06-30 00:29:58
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 96 ms / 2,000 ms
コード長 1,373 bytes
コンパイル時間 961 ms
コンパイル使用メモリ 86,416 KB
実行使用メモリ 57,916 KB
最終ジャッジ日時 2023-09-26 11:29:36
合計ジャッジ時間 3,841 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,380 KB
testcase_01 AC 2 ms
4,380 KB
testcase_02 AC 2 ms
4,380 KB
testcase_03 AC 64 ms
40,504 KB
testcase_04 AC 2 ms
4,380 KB
testcase_05 AC 54 ms
33,236 KB
testcase_06 AC 95 ms
57,832 KB
testcase_07 AC 14 ms
9,772 KB
testcase_08 AC 30 ms
19,216 KB
testcase_09 AC 72 ms
43,932 KB
testcase_10 AC 78 ms
47,680 KB
testcase_11 AC 79 ms
47,812 KB
testcase_12 AC 92 ms
54,968 KB
testcase_13 AC 22 ms
14,600 KB
testcase_14 AC 41 ms
26,420 KB
testcase_15 AC 67 ms
41,140 KB
testcase_16 AC 54 ms
33,756 KB
testcase_17 AC 58 ms
35,356 KB
testcase_18 AC 79 ms
48,812 KB
testcase_19 AC 68 ms
41,252 KB
testcase_20 AC 29 ms
18,780 KB
testcase_21 AC 71 ms
42,800 KB
testcase_22 AC 40 ms
25,144 KB
testcase_23 AC 96 ms
57,828 KB
testcase_24 AC 96 ms
57,916 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <iostream>
#include <map>
#include <queue>
#include <deque>
#include <string>
#include <stack>
#include <vector>
#include <set>
#include <tuple>
#include <utility>
#include <functional>
using namespace std;
typedef long long ll;
typedef pair<int,int> P;
typedef tuple<int,int,int> T;
const int INF = 1000000000;
const int MOD = 1000000007;
int main(){
	ll p,k;
	cin >> p >> k;
	vector<vector<ll>> dp(k+1,vector<ll>(2,0));
	dp[0][0] = 1;
	for(ll i = 0;i < k;i++){
		//+について
		{
			//0->0は0を選ぶしか無い.
			dp[i+1][0] += dp[i][0];
			dp[i+1][0] %= MOD;

			//1->0は選んで0となるのが唯一存在する.
			dp[i+1][0] += dp[i][1];
			dp[i+1][0] %= MOD;

			//0->1は1通りずつ存在するので,まとめ上げるとp-1倍.
			dp[i+1][1] += (p-1) * dp[i][0];
			dp[i+1][1] %= MOD;

			//1->1は1通りずつ存在するので,まとめ上げるとp-1倍.
			dp[i+1][1] += (p-1) * dp[i][1];
			dp[i+1][1] %= MOD;
		}

		//xについて
		{
			//0->0はどれを選んでもよい.
			dp[i+1][0] += p * dp[i][0];
			dp[i+1][0] %= MOD;
			//1->0は0を選べば得られる.
			dp[i+1][0] += dp[i][1];
			dp[i+1][0] %= MOD;

			//0->1はなし.
			//1->1は+と同様.
			dp[i+1][1] += (p -1 ) * dp[i][1];
			dp[i+1][1] %= MOD;
		}
	}

	cout << dp[k][0] << endl;
}
0