結果

問題 No.1049 Zero (Exhaust)
ユーザー izryt(趣味)
提出日時 2020-05-28 19:25:42
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 17 ms / 2,000 ms
コード長 1,486 bytes
コンパイル時間 1,060 ms
コンパイル使用メモリ 92,640 KB
最終ジャッジ日時 2025-01-10 16:06:16
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <cstdio>
#include <string>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <utility>
#include <functional>
#include <climits>
#include <cstring>
#include <unordered_map>

using namespace std;

#define int long long
#define rep(i, n) for (int i=0;i<(int)(n);++i)
#define rep1(i, n) for (int i=1;i<=(int)(n);++i)
#define range(i, l, r) for (int i=l;i<(int)(r);++i)
#define rrange(i, l, r) for (int i=r-1;i>=(int)(l);--i)
#define unless(a) if(!(a))
#define all(a) begin(a),end(a)
#define fst first
#define scd second
#define PB emplace_back
#define PPB pop_back

using vi=vector<int>;
using pii=pair<int, int>;
using ll=long long;

bool chmin(int&a,int b){return a>b?(a=b,true):false;}
bool chmax(int&a,int b){return a<b?(a=b,true):false;}
//int read(){int a;scanf("%lld",&a);return a;}

constexpr int TEN(int n){return n==0?1:10*TEN(n-1);}

const int inf = TEN(9) + 10;
const int mod = TEN(9) + 7;

int P, K;
int add[2], mul[2];

signed main()
{
	cin >> P >> K;
	add[0] = 1, mul[0] = 0;
	rep(i, K) {
		int nadd[2] = {0, 0}, nmul[2] = {0, 0};
		int zero = add[0] + mul[0], non_zero = add[1] + mul[1];

		nadd[0] += zero + non_zero;
		nadd[1] += zero * (P - 1) % mod + non_zero * (P - 1) % mod;

		nmul[0] += zero * P % mod + non_zero;
		nmul[1] += non_zero * (P - 1) % mod;

		rep(j, 2) {
			add[j] = nadd[j] % mod;
			mul[j] = nmul[j] % mod;
		}
	}

	int ans = (add[0] + mul[0]) % mod;

	cout << ans << endl;
}
0