結果
問題 | No.1050 Zero (Maximum) |
ユーザー |
![]() |
提出日時 | 2020-05-14 18:25:32 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 24 ms / 2,000 ms |
コード長 | 1,549 bytes |
コンパイル時間 | 3,049 ms |
コンパイル使用メモリ | 198,888 KB |
最終ジャッジ日時 | 2025-01-10 10:54:21 |
ジャッジサーバーID (参考情報) |
judge3 / judge6 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 15 |
ソースコード
#include <bits/stdc++.h> #define rep(i,n) for(int i = 0; i < (n); ++i) #define srep(i,s,t) for (int i = s; i < t; ++i) #define drep(i,n) for(int i = (n)-1; i >= 0; --i) using namespace std; typedef long long int ll; typedef pair<int,int> P; #define yn {puts("Yes");}else{puts("No");} #define MAX_N 200005 const ll MOD = 1000000007; int m, k; // 遷移行列のサイズ // DPの更新 vector<ll> matmul(vector<ll> &dp, vector<vector<ll>> &mt){ vector<ll> ret(m,0); rep(i,m){ rep(j,m){ ret[i] += mt[i][j] * dp[j] % MOD; ret[i] %= MOD; } } return ret; } // 遷移行列の更新 vector<vector<ll>> update_matmul(vector<vector<ll>> &mt){ vector<vector<ll>> ret(m, vector<ll>(m,0)); rep(i,m){ rep(j,m){ rep(k,m){ ret[i][j] += mt[i][k] * mt[k][j] % MOD; ret[i][j] %= MOD; } } } return ret; } void matpow(vector<ll> &dp, vector<vector<ll>> &mt, ll k){ int m = dp.size(); while(k){ if(k & 1) dp = matmul(dp, mt); mt = update_matmul(mt); k /= 2; } } int main() { cin >> m >> k; vector<ll> dp(m); rep(i,m) dp[i] = 0; dp[0] = 1; vector<vector<ll>> mt(m,vector<ll>(m)); rep(i,m)rep(j,m)mt[i][j] = 0; rep(i,m){ rep(j,m){ mt[(j+i)%m][j]++; } } rep(i,m){ rep(j,m){ mt[(j*i)%m][j]++; } } matpow(dp, mt, k); ll ans = dp[0]; cout << ans << endl; return 0; }