結果
問題 | No.1847 Good Sequence |
ユーザー |
![]() |
提出日時 | 2022-02-18 22:42:39 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 49 ms / 3,000 ms |
コード長 | 2,137 bytes |
コンパイル時間 | 2,519 ms |
コンパイル使用メモリ | 203,492 KB |
最終ジャッジ日時 | 2025-01-28 00:25:51 |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 41 |
ソースコード
#include <bits/stdc++.h> using namespace std; const long long MOD = 1000000007; long long modpow(long long a, long long b){ long long ans = 1; while (b > 0){ if (b % 2 == 1){ ans *= a; ans %= MOD; } a *= a; a %= MOD; b /= 2; } return ans; } vector<vector<long long>> matmul(vector<vector<long long>> A, vector<vector<long long>> B){ int N = A.size(); vector<vector<long long>> ans(N, vector<long long>(N, 0)); for (int i = 0; i < N; i++){ for (int j = 0; j < N; j++){ for (int k = 0; k < N; k++){ ans[i][k] += A[i][j] * B[j][k]; ans[i][k] %= MOD; } } } return ans; } vector<vector<long long>> matexp(vector<vector<long long>> A, long long b){ int N = A.size(); vector<vector<long long>> ans(N, vector<long long>(N, 0)); for (int i = 0; i < N; i++){ ans[i][i] = 1; } while (b > 0){ if (b % 2 == 1){ ans = matmul(ans, A); } A = matmul(A, A); b /= 2; } return ans; } int main(){ long long L; int N, M; cin >> L >> N >> M; vector<int> K(M); for (int i = 0; i < M; i++){ cin >> K[i]; } long long ans = modpow(N, L); vector<int> id(N + 1, -1); for (int i = 0; i < M; i++){ id[K[i]] = i; } vector<int> S(M + 1); S[0] = 1; for (int i = 0; i < M; i++){ S[i + 1] = S[i] + K[i] + 1; } vector<vector<long long>> A(S[M], vector<long long>(S[M], 0)); A[0][0] += N - M; for (int i = 0; i < M; i++){ A[0][S[i]]++; } for (int i = 0; i < M; i++){ for (int j = 1; j <= K[i] + 1; j++){ for (int k = 0; k < N; k++){ int p = S[i] + j - 1; if (K[i] == k + 1){ if (j <= K[i]){ A[p][p + 1]++; } else { A[p][p]++; } } else if (id[k + 1] != -1){ if (j != K[i]){ A[p][S[id[k + 1]]]++; } } else { if (j != K[i]){ A[p][0]++; } } } } } A = matexp(A, L); ans += MOD - A[0][0]; for (int i = 0; i < M; i++){ for (int j = 1; j <= K[i] + 1; j++){ if (j != K[i]){ ans += MOD - A[0][S[i] + j - 1]; } } } ans %= MOD; cout << ans << endl; }