結果

問題 No.2367 Painting Gascket
ユーザー SSRSSSRS
提出日時 2023-06-30 22:46:56
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
RE  
実行時間 -
コード長 1,930 bytes
コンパイル時間 1,530 ms
コンパイル使用メモリ 172,460 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-07-07 10:38:52
合計ジャッジ時間 5,624 ms
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

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

ソースコード

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);
  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] * dp112[i] % MOD * dp111[i] % MOD;
      dp112[i + 1] += dp112[i] * 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