結果

問題 No.1112 冥界の音楽
ユーザー momoyuumomoyuu
提出日時 2022-08-18 18:18:15
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 93 ms / 2,000 ms
コード長 1,878 bytes
コンパイル時間 3,144 ms
コンパイル使用メモリ 255,072 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-10-06 12:06:16
合計ジャッジ時間 4,127 ms
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff #

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

using ll = long long;
using ull = unsigned long long;
using ld = long double;
template<class t> using vc = vector<t>;
template<class t> using vvc = vc<vc<t>>;
using pi = pair<int,int>;
using vi = vc<int>;
using vvi = vvc<int>;

#define rep(i,a,b) for (int i = a; i < b; i++)
#define irep(i,a,b) for (int i = a; i > b; i--)
#define print(n) cout << n << endl
#define rup(a,b) (a+b-1)/b
#define input(A,N) rep(i,0,N) cin>>A[i];

vvc<ll> mattimes(vvc<ll> A,vvc<ll> B,ll mod = 1){
    vvc<ll> now(A.size(),vc<ll>(A.size(),0));
    rep(i,0,A.size()){
        rep(j,0,A.size()){
            rep(k,0,A.size()){
                now[i][j] += (A[i][k]*B[k][j])%mod;
                now[i][j] %= mod;
            }
        }
    }
    return now;
}

vc<ll> mattimes(vvc<ll> A,vc<ll> B,ll mod = 1){
    vc<ll> now(B.size(),0);
    rep(i,0,B.size()){
        rep(j,0,B.size()){
            now[i] += (A[i][j]*B[j])%mod;
            now[i] %= mod;
        }
    }
    return now;
}
int main(){
    cout << fixed << setprecision(15);
    
    int K,M;
    ll N;
    cin>>K>>M>>N;
    int P[M],Q[M],R[M];
    rep(i,0,M) {cin>>P[i]>>Q[i]>>R[i];P[i]--;Q[i]--;R[i]--;}

    vvc<ll> mat(K*K,vc<ll>(K*K,0));
    rep(i,0,M){
        int p,q;
        p = P[i]*K + Q[i];
        q = Q[i]*K + R[i];
        mat[q][p] = 1;
    }
    ll mod = ll(pow(10,9)) + 7;
    vvc<ll> A(K*K,vc<ll>(K*K,0));
    rep(i,0,K*K) A[i][i] = 1;
    int now = 0;
    while (pow(2,now)<=N-2){
        if (N-2>>now & 1){
            A = mattimes(A,mat,mod);
        }
        mat = mattimes(mat,mat,mod);
        now ++ ;
    }

    vc<ll> ans(K*K,0);
    rep(i,0,K){
        ans[i] = 1;
    }
    ans = mattimes(A,ans,mod);

    ull count = 0;
    rep(i,0,K){
        count += ans[i*K];
        count %= mod;
    }

    print(count);

    
    //system("pause");
    return 0;
}
0