結果
問題 | No.2367 Painting Gascket |
ユーザー |
|
提出日時 | 2023-07-01 01:06:00 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 8 ms / 2,000 ms |
コード長 | 867 bytes |
コンパイル時間 | 619 ms |
コンパイル使用メモリ | 69,440 KB |
実行使用メモリ | 7,104 KB |
最終ジャッジ日時 | 2024-07-07 12:19:34 |
合計ジャッジ時間 | 1,743 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 34 |
ソースコード
#include<iostream> #include<cassert> #include<atcoder/modint> using namespace std; using mint=atcoder::modint1000000007; int K,N; mint dp[1<<17][7]; int main() { ios::sync_with_stdio(false); cin.tie(nullptr); cin>>K>>N; dp[0][0]=N; dp[0][1]=N-1; dp[0][2]=N-1; dp[0][3]=N-1; dp[0][4]=N-2; dp[0][5]=N-2; dp[0][6]=N-3; for(int i=1;i<=K;i++) { dp[i][0]=N*dp[i-1][1].pow(3); dp[i][1]=dp[i-1][2].pow(2)*dp[i-1][1]+(N-1)*dp[i-1][4].pow(2)*dp[i-1][1]; dp[i][2]=dp[i-1][3]*dp[i-1][2].pow(2)+(N-1)*dp[i-1][5]*dp[i-1][4].pow(2); dp[i][3]=dp[i-1][3].pow(3)+(N-1)*dp[i-1][5].pow(3); dp[i][4]=2*dp[i-1][5]*dp[i-1][2]*dp[i-1][4]+(N-2)*dp[i-1][6]*dp[i-1][4].pow(2); dp[i][5]=dp[i-1][3]*dp[i-1][5].pow(2)+dp[i-1][5].pow(3)+(N-2)*dp[i-1][5]*dp[i-1][6].pow(2); dp[i][6]=3*dp[i-1][5].pow(2)*dp[i-1][6]+(N-3)*dp[i-1][6].pow(3); } cout<<dp[K][0].val()<<endl; }