結果

問題 No.1112 冥界の音楽
コンテスト
ユーザー momoyuu
提出日時 2022-08-18 18:20:34
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 103 ms / 2,000 ms
コード長 1,836 bytes
コンパイル時間 3,556 ms
コンパイル使用メモリ 254,096 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-06 12:09:49
合計ジャッジ時間 4,954 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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;
//#define int ll
template<class t> using vc = vector<t>;
template<class t> using vvc = vc<vc<t>>;
using pi = pair<int,int>;
using vi = vc<ll>;
using vvi = vvc<ll>;

#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];

vvi mattimes(vvi A,vvi B,ll mod = 1){
    vvi now(A.size(),vi(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;
}

vi mattimes(vvi A,vi B,ll mod = 1){
    vi 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]--;}

    vvi mat(K*K,vi(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;
    vvi A(K*K,vi(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 ++ ;
    }

    vi 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