結果
問題 | No.1112 冥界の音楽 |
ユーザー |
![]() |
提出日時 | 2020-07-10 21:46:22 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 54 ms / 2,000 ms |
コード長 | 2,433 bytes |
コンパイル時間 | 4,801 ms |
コンパイル使用メモリ | 206,368 KB |
最終ジャッジ日時 | 2025-01-11 18:15:52 |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 34 |
ソースコード
#include <bits/stdc++.h> using namespace std; #define modulo 1000000007 #define mod(mod_x) ((((long long)mod_x+modulo))%modulo) #define Inf 1000000000 int beki(int a,int b){ int x = 1; while(b!=0){ if(b&1){ x=mod(x*a); } a=mod(a*a); b>>=1; } return x; } struct matrix{ vector<vector<long long>> value; matrix(int height,int width){ value.resize(height,vector<long long>(width,0)); if(height==width){ for(int i=0;i<get_width();i++){ value[i][i] = 1; } } } void resize(int height,int width){ value.resize(height,vector<long long>(width,0)); for(int i=0;i<get_width();i++){ value[i][i] = 1; } } void set_value(vector<vector<long long>> &V){ value = V; } void set_value(initializer_list<long long> list){ auto it = list.begin(); for(int i=0;i<get_height();i++){ for(int j=0;j<get_width();j++){ value[i][j]=*it; if(i!=get_height()-1||j!=get_width()-1)it++; } } } int get_width(){ return value[0].size(); } int get_height(){ return value.size(); } void print(){ for(int i=0;i<get_height();i++){ for(int j=0;j<get_width();j++){ if(j!=0)cout<<' '; cout<<value[i][j]; } cout<<endl; } } }; matrix matrix_multiple(matrix &A,matrix &B){ matrix R(A.get_height(),B.get_width()); int multi_cnt; if(A.get_width()!=B.get_height()){ assert(false); } else multi_cnt = A.get_width(); for(int i=0;i<R.get_height();i++){ for(int j=0;j<R.get_width();j++){ R.value[i][j]=0; for(int k=0;k<multi_cnt;k++){ R.value[i][j] = mod(R.value[i][j] + mod(A.value[i][k] * B.value[k][j])); } } } return R; } matrix matrix_beki(matrix A,long long cnt){ if(A.get_width()!=A.get_height())assert(0); matrix R(A.get_width(),A.get_width()); while(cnt!=0){ if(cnt&1){ R=matrix_multiple(R,A); } A=matrix_multiple(A,A); cnt>>=1; } return R; } int main(){ int K,MM; cin>>K>>MM; long long N; cin>>N; vector<vector<long long>> T(K*K,vector<long long>(K*K,0LL)); for(int i=0;i<MM;i++){ int p,q,r; cin>>p>>q>>r; p--,q--,r--; T[p*K+q][q*K+r]++; } matrix M(K*K,K*K); M.set_value(T); M = matrix_beki(M,N-2); vector<vector<long long>> X(1,vector<long long>(K*K,0LL)); for(int i=0;i<K;i++)X[0][i]=1; matrix A(1,K*K); A.set_value(X); A = matrix_multiple(A,M); int ans = 0; for(int i=0;i<K*K;i++){ if(i%K==0)ans = mod(ans + A.value[0][i]); } cout<<ans<<endl; return 0; }