結果

問題 No.2367 Painting Gascket
ユーザー SSRSSSRS
提出日時 2023-06-30 22:35:16
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
RE  
実行時間 -
コード長 1,952 bytes
コンパイル時間 1,890 ms
コンパイル使用メモリ 168,632 KB
実行使用メモリ 8,712 KB
最終ジャッジ日時 2023-09-21 16:44:29
合計ジャッジ時間 5,173 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 RE -
testcase_01 RE -
testcase_02 WA -
testcase_03 RE -
testcase_04 RE -
testcase_05 RE -
testcase_06 RE -
testcase_07 RE -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 RE -
testcase_30 RE -
testcase_31 RE -
testcase_32 RE -
testcase_33 RE -
testcase_34 RE -
testcase_35 RE -
testcase_36 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
const long long MOD = 1000000007;
int main(){
  int K;
  long long N;
  cin >> K >> N;
  assert(K >= 3 && N >= 4);
  vector<long long> dp000(K + 1, 0);
  vector<long long> dp001(K + 1, 0);
  vector<long long> dp011(K + 1, 0);
  vector<long long> dp012(K + 1, 0);
  vector<long long> dp111(K + 1, 0);
  vector<long long> dp112(K + 1, 0);
  vector<long long> dp123(K + 1, 0);
  dp000[0] = N;
  dp001[0] = N - 1;
  dp011[0] = N - 1;
  dp012[0] = max(N - 2, (long long) 0);
  dp111[0] = N - 1;
  dp112[0] = max(N - 2, (long long) 0);
  dp123[0] = max(N - 3, (long long) 0);
  for (int i = 0; i < K; i++){
    dp000[i + 1] = N * dp001[i] % MOD * dp001[i] % MOD * dp001[i] % MOD;
    dp001[i + 1] += (N - 1) * dp001[i] % MOD * dp012[i] % MOD * dp012[i] % MOD;
    dp001[i + 1] += dp001[i] * dp011[i] % MOD * dp011[i] % MOD;
    dp001[i + 1] %= MOD;
    dp011[i + 1] += (N - 1) * dp012[i] % MOD * dp012[i] % MOD * dp112[i] % MOD;
    dp011[i + 1] += dp011[i] * dp011[i] % MOD * dp111[i] % MOD;
    dp011[i + 1] %= MOD;
    if (N >= 2){
      dp012[i + 1] += (N - 2) * dp012[i] % MOD * dp012[i] % MOD * dp123[i] % MOD;
      dp012[i + 1] += 2 * dp011[i] % MOD * dp012[i] % MOD * dp112[i] % MOD;
      dp012[i + 1] %= MOD;
    }
    dp111[i + 1] += (N - 1) * dp112[i] % MOD * dp112[i] % MOD * dp112[i] % MOD;
    dp111[i + 1] += dp111[i] * dp111[i] % MOD * dp111[i] % MOD;
    dp111[i + 1] %= MOD;
    if (N >= 2){
      dp112[i + 1] += (N - 2) * dp112[i] % MOD * dp123[i] % MOD * dp123[i] % MOD;
      dp112[i + 1] += dp112[i] % MOD * dp112[i] % MOD * dp111[i] % MOD;
      dp112[i + 1] += dp112[i] % MOD * dp112[i] % MOD * dp112[i] % MOD;
      dp112[i + 1] %= MOD;
    }
    if (N >= 3){
      dp123[i + 1] += (N - 3) * dp123[i] % MOD * dp123[i] % MOD * dp123[i] % MOD;
      dp123[i + 1] += 3 * dp112[i] % MOD * dp123[i] % MOD * dp123[i] % MOD;
      dp123[i + 1] %= MOD;
    }
  }
  cout << dp000[K] << endl;
}
0