#836077 (C++14) No.621 3 x N グリッド上のドミノの置き方の数

提出ソース

結果

問題	No.621 3 x N グリッド上のドミノの置き方の数
ユーザー	daddy
提出日時	2023-01-29 23:01:21
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	12 ms / 3,000 ms
コード長	3,459 bytes
コンパイル時間	1,822 ms
コンパイル使用メモリ	180,072 KB
実行使用メモリ	6,944 KB
最終ジャッジ日時	2024-06-29 18:02:05
合計ジャッジ時間	4,034 ms
ジャッジサーバーID （参考情報）	judge5 / judge3

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	AC	2 ms 5,376 KB
testcase_02	AC	2 ms 5,376 KB
testcase_03	AC	2 ms 5,376 KB
testcase_04	AC	3 ms 5,376 KB
testcase_05	AC	3 ms 5,376 KB
testcase_06	AC	2 ms 5,376 KB
testcase_07	AC	2 ms 5,376 KB
testcase_08	AC	2 ms 5,376 KB
testcase_09	AC	3 ms 5,376 KB
testcase_10	AC	3 ms 5,376 KB
testcase_11	AC	2 ms 5,376 KB
testcase_12	AC	3 ms 5,376 KB
testcase_13	AC	3 ms 5,376 KB
testcase_14	AC	4 ms 5,376 KB
testcase_15	AC	3 ms 5,376 KB
testcase_16	AC	4 ms 5,376 KB
testcase_17	AC	4 ms 5,376 KB
testcase_18	AC	5 ms 5,376 KB
testcase_19	AC	5 ms 5,376 KB
testcase_20	AC	6 ms 5,376 KB
testcase_21	AC	6 ms 5,376 KB
testcase_22	AC	7 ms 5,376 KB
testcase_23	AC	7 ms 5,376 KB
testcase_24	AC	7 ms 6,944 KB
testcase_25	AC	8 ms 6,940 KB
testcase_26	AC	8 ms 6,940 KB
testcase_27	AC	9 ms 6,940 KB
testcase_28	AC	8 ms 6,940 KB
testcase_29	AC	10 ms 6,940 KB
testcase_30	AC	9 ms 6,940 KB
testcase_31	AC	10 ms 6,944 KB
testcase_32	AC	9 ms 6,944 KB
testcase_33	AC	10 ms 6,940 KB
testcase_34	AC	9 ms 6,944 KB
testcase_35	AC	10 ms 6,944 KB
testcase_36	AC	10 ms 6,944 KB
testcase_37	AC	11 ms 6,944 KB
testcase_38	AC	10 ms 6,944 KB
testcase_39	AC	10 ms 6,940 KB
testcase_40	AC	10 ms 6,944 KB
testcase_41	AC	10 ms 6,944 KB
testcase_42	AC	10 ms 6,944 KB
testcase_43	AC	10 ms 6,940 KB
testcase_44	AC	10 ms 6,940 KB
testcase_45	AC	10 ms 6,944 KB
testcase_46	AC	10 ms 6,940 KB
testcase_47	AC	11 ms 6,940 KB
testcase_48	AC	10 ms 6,940 KB
testcase_49	AC	10 ms 6,940 KB
testcase_50	AC	10 ms 6,940 KB
testcase_51	AC	10 ms 6,940 KB
testcase_52	AC	10 ms 6,940 KB
testcase_53	AC	10 ms 6,944 KB
testcase_54	AC	11 ms 6,940 KB
testcase_55	AC	10 ms 6,944 KB
testcase_56	AC	11 ms 6,940 KB
testcase_57	AC	8 ms 6,944 KB
testcase_58	AC	9 ms 6,944 KB
testcase_59	AC	12 ms 6,940 KB
testcase_60	AC	12 ms 6,944 KB
testcase_61	AC	12 ms 6,940 KB
testcase_62	AC	7 ms 6,940 KB
testcase_63	AC	7 ms 6,940 KB
testcase_64	AC	8 ms 6,940 KB
testcase_65	AC	8 ms 6,940 KB
testcase_66	AC	7 ms 6,940 KB
testcase_67	AC	7 ms 6,940 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#pragma GCC optimization ("O3")

#include <bits/stdc++.h>

using namespace std;

using ll = long long;
using vec = vector<ll>;
using mat = vector<vec>;
using pll = pair<ll,ll>;

#define INF (1LL<<61)
#define MOD 1000000007LL 
//#define MOD 998244353LL
#define EPS (1e-10)

#define PR(x) cout << (x) << endl
#define PS(x) cout << (x) << " "
#define REP(i,m,n) for(ll (i)=(m),(i_len)=(n);(i)<(i_len);++(i))
#define FORE(i,v) for(auto (i):v)
#define ALL(x) (x).begin(), (x).end()
#define SZ(x) ((ll)(x).size())
#define REV(x) reverse(ALL((x)))
#define ASC(x) sort(ALL((x)))
#define DESC(x) {ASC((x)); REV((x));}
#define BIT(s,i) (((s)>>(i))&1)
#define pb push_back
#define fi first
#define se second

template<class T> inline int chmin(T& a, T b) {if(a>b) {a=b; return 1;} return 0;}
template<class T> inline int chmax(T& a, T b) {if(a<b) {a=b; return 1;} return 0;}
class mint {
public:
    ll x;
    mint(ll x=0) : x((x%MOD+MOD)%MOD) {}
    mint operator-() const {return mint(-x);}
    mint& operator+=(const mint& a) {if((x+=a.x)>=MOD) x-=MOD; return *this;}
    mint& operator-=(const mint& a) {if((x+=MOD-a.x)>=MOD) x-=MOD; return *this;}
    mint& operator*=(const mint& a) {(x*=a.x)%=MOD; return *this;}
    mint operator+(const mint& a) const {mint b(*this); return b+=a;}
    mint operator-(const mint& a) const {mint b(*this); return b-=a;}
    mint operator*(const mint& a) const {mint b(*this); return b*=a;}
    mint pow(ll t) const {if(!t) return 1; mint a=pow(t>>1); return (t&1?*this*a:a)*a;}
    mint inv() const {return pow(MOD-2);}
    mint& operator/=(const mint& a) {return *this*=a.inv();}
    mint operator/(const mint& a) const {mint b(*this); return b/=a;}
};
istream &operator>>(istream& is, mint& a) {ll t; is>>t; a=t; return is;}
ostream &operator<<(ostream& os, const mint& a) {return os<<a.x;}
using mvec = vector<mint>;
using mmat = vector<mvec>;

mat matmul(mat A, mat B)
{
    ll N = SZ(A);
    mat C(N, vec(N, 0));
    REP(i,0,N) {
        REP(j,0,N) {
            REP(k,0,N) C[i][j] += A[i][k]*B[k][j], C[i][j] %= MOD;
        }
    }
    return C;
}

mat matpow(mat A, ll n)
{
    if(n == 0) {
        ll N = SZ(A);
        mat I(N, vec(N, 0));
        REP(i,0,N) I[i][i] = 1;
        return I;
    }
    mat T = matpow(A, n>>1);
    T = matmul(T, T);
    if(n&1) T = matmul(T, A);
    return T; 
}

int main()
{
    ll N;
    cin >> N;

    string S = "XILR";
    vector<string> T;
    REP(i,0,4) {
        REP(j,0,4) {
            REP(k,0,4) {
                string t = "***";
                if((i == 0 && j == 0) || (j == 0 && k == 0)) continue;
                if(!((i == 1 && j == 1 && k != 1) || (i != 1 && j == 1 && k == 1) || (i != 1 && j != 1 && k != 1))) continue;
                t[0] = S[i]; t[1] = S[j]; t[2] = S[k];
                T.pb(t);
            }
        }
    }

    auto check = [](string x, string y) {
        bool res = true;
        REP(i,0,3) {
            bool tmp = false;
            tmp |= (x[i] == 'X' && (y[i] == 'I' || y[i] == 'L'));
            tmp |= (x[i] == 'I' && (y[i] == 'X' || y[i] == 'I' || y[i] == 'L'));
            tmp |= (x[i] == 'L' && y[i] == 'R');
            tmp |= (x[i] == 'R' && (y[i] == 'X' || y[i] == 'I' || y[i] == 'L'));
            res &= tmp;
        }
        return res;
    };

    ll M = SZ(T);
    mat A(M, vec(M));
    REP(i,0,M) {
        REP(j,0,M) A[i][j] = check(T[i], T[j]);
    }
    A = matpow(A, N+1);
    PR(A[27][14]);

    return 0;
}

/*



*/

yukicoder

結果

テストケース

ソースコード