結果

問題 No.2367 Painting Gascket
コンテスト
ユーザー hitonanode
提出日時 2023-06-30 22:37:27
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 6 ms / 2,000 ms
コード長 1,024 bytes
コンパイル時間 819 ms
コンパイル使用メモリ 86,748 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-07-07 10:28:11
合計ジャッジ時間 1,817 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC optimize("O3")
#include <iostream>
using namespace std;
#define FOR(i, begin, end) for(int i=(begin),i##_end_=(end);i<i##_end_;i++)
#define REP(i, n) FOR(i,0,n)

#include <atcoder/modint>
using mint = atcoder::modint1000000007;

int main() {
    cin.tie(nullptr), ios::sync_with_stdio(false);

    int K, N;
    cin >> K >> N;

    if (K == 0) {
        cout << N << '\n';
        return 0;
    }

    const int M = max(0, N - 2), U = max(0, N - 3);

    mint g = N - 1, h = N - 1, i = M, j = N - 1, k = M, l = U;

    REP(_, K - 1) {
        mint gnxt = g * (h * h + i * i * (N - 1));
        mint hnxt = j * h * h + k * i * i * (N - 1);
        mint inxt = h * k * i * 2 + l * i * i * M;
        mint jnxt = j * j * j + k * k * k * (N - 1);
        mint knxt = j * k * k + k * k * k + k * l * l * M;
        mint lnxt = l * k * k * 3 + l * l * l * U;
        g = gnxt;
        h = hnxt;
        i = inxt;
        j = jnxt;
        k = knxt;
        l = lnxt;
    }
    cout << (g * g * g * N).val() << '\n';
}
0