結果

問題 No.541 3 x N グリッド上のサイクルの個数
ユーザー Aquarius
提出日時 2020-03-15 21:54:28
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 1,294 bytes
コンパイル時間 1,630 ms
コンパイル使用メモリ 175,620 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-25 22:25:20
合計ジャッジ時間 3,255 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 62
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

const long mod = 1000000007;
using Mat = vector<vector<long>>;

int size = 10;
Mat mat = {
    { 1, 1, 1, 1, 1, 1, 1, 1, 0, 0 },
    { 0, 1, 1, 1, 0, 0, 0, 1, 0, 1 },
    { 0, 1, 1, 1, 1, 1, 0, 0, 0, 1 },
    { 0, 1, 1, 1, 1, 1, 1, 0, 1, 1 },
    { 0, 0, 1, 1, 1, 1, 0, 0, 0, 1 },
    { 0, 0, 1, 1, 1, 1, 1, 0, 0, 1 },
    { 0, 0, 0, 1, 0, 1, 1, 1, 0, 1 },
    { 0, 0, 0, 1, 0, 0, 0, 1, 0, 0 },
    { 0, 1, 0, 0, 0, 0, 1, 0, 1, 1 },
    { 0, 0, 0, 0, 0, 0, 0, 0, 0, 1 }
};

Mat zero() {
    return Mat(size, vector<long>(size));
}

Mat id() {
    Mat ret = zero();
    for (int i = 0; i < size; ++i) {
        ret[i][i] = 1;
    }
    return ret;
}

Mat mul(Mat a, Mat b) {
    Mat ret = zero();
    for (int i = 0; i < size; ++i) {
        for (int k = 0; k < size; ++k) {
            for (int j = 0; j < size; ++j) {
                ret[i][j] = (ret[i][j] + a[i][k] * b[k][j]) % mod;
            }
        }
    }
    return ret;
}

Mat power(Mat a, long ex) {
    Mat ret = id();
    Mat base = a;
    while (ex > 0) {
        if (ex & 1) ret = mul(ret, base);
        base = mul(base, base);
        ex >>= 1;
    }
    return ret;
}

long n;
int main() {
    cin >> n;
    auto res = power(mat, n + 1);
    cout << res[0][size - 1] << endl;
}
0