結果
問題 | No.1112 冥界の音楽 |
ユーザー |
|
提出日時 | 2020-07-10 21:47:30 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 14 ms / 2,000 ms |
コード長 | 1,203 bytes |
コンパイル時間 | 2,075 ms |
コンパイル使用メモリ | 203,532 KB |
最終ジャッジ日時 | 2025-01-11 18:16:44 |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 34 |
ソースコード
#include <bits/stdc++.h> using namespace std; const int64_t MOD = 1e9+7; void add(int64_t& a, int64_t b){ a = (a+b) % MOD; } void mul(int64_t& a, int64_t b){ a = a*b % MOD; } typedef vector<vector<int64_t>> Mat; Mat matmul(Mat A, Mat B){ assert(A[0].size() == B.size()); int N = A.size(), M = B[0].size(), K = B.size(); Mat ans(N, vector<int64_t>(M)); for(int i=0; i<N; i++) for(int j=0; j<M; j++) for(int k=0; k<K; k++) add(ans[i][j], A[i][k] * B[k][j]); return ans; } int main(){ int K, M; int64_t N; cin >> K >> M >> N; auto encode = [&](int i, int j){ return i*K+j; }; vector<int> P(M), Q(M), R(M); for(int i=0; i<M; i++){ cin >> P[i] >> Q[i] >> R[i]; P[i]--; Q[i]--; R[i]--; } int S = K*K; Mat X(S, vector<int64_t>(S)), A(S, vector<int64_t>(1)); for(int i=0; i<M; i++){ X[encode(Q[i], R[i])][encode(P[i], Q[i])]++; } for(int i=0; i<K; i++) A[encode(0, i)][0]++; N -= 2; while(N){ if(N%2) A = matmul(X, A); N /= 2; X = matmul(X, X); } int64_t ans = 0; for(int i=0; i<K; i++) add(ans, A[encode(i, 0)][0]); cout << ans << endl; return 0; }