結果
問題 | No.840 ほむほむほむら |
ユーザー | ok |
提出日時 | 2019-06-25 19:01:12 |
言語 | C++11 (gcc 11.4.0) |
結果 |
AC
|
実行時間 | 362 ms / 4,000 ms |
コード長 | 1,776 bytes |
コンパイル時間 | 936 ms |
コンパイル使用メモリ | 85,044 KB |
実行使用メモリ | 4,380 KB |
最終ジャッジ日時 | 2023-09-05 09:19:02 |
合計ジャッジ時間 | 4,972 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge15 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
4,380 KB |
testcase_01 | AC | 2 ms
4,376 KB |
testcase_02 | AC | 8 ms
4,380 KB |
testcase_03 | AC | 50 ms
4,376 KB |
testcase_04 | AC | 1 ms
4,376 KB |
testcase_05 | AC | 1 ms
4,376 KB |
testcase_06 | AC | 2 ms
4,376 KB |
testcase_07 | AC | 16 ms
4,376 KB |
testcase_08 | AC | 91 ms
4,380 KB |
testcase_09 | AC | 2 ms
4,376 KB |
testcase_10 | AC | 2 ms
4,376 KB |
testcase_11 | AC | 4 ms
4,380 KB |
testcase_12 | AC | 25 ms
4,380 KB |
testcase_13 | AC | 235 ms
4,380 KB |
testcase_14 | AC | 30 ms
4,376 KB |
testcase_15 | AC | 2 ms
4,380 KB |
testcase_16 | AC | 5 ms
4,380 KB |
testcase_17 | AC | 53 ms
4,376 KB |
testcase_18 | AC | 297 ms
4,376 KB |
testcase_19 | AC | 362 ms
4,376 KB |
testcase_20 | AC | 1 ms
4,376 KB |
testcase_21 | AC | 2 ms
4,380 KB |
testcase_22 | AC | 6 ms
4,376 KB |
testcase_23 | AC | 347 ms
4,376 KB |
testcase_24 | AC | 5 ms
4,376 KB |
testcase_25 | AC | 1 ms
4,376 KB |
testcase_26 | AC | 8 ms
4,376 KB |
testcase_27 | AC | 346 ms
4,376 KB |
ソースコード
#include<iostream> #include<string> #include<iomanip> #include<cmath> #include<vector> #include<algorithm> using namespace std; #define int long long #define rep(i,n) for(int i = 0; i < (n); i++) #define endl "\n" const long long INF = (long long)1e18; const long long MOD = (long long)998244353; string yn(bool f){return f?"Yes":"No";} string YN(bool f){return f?"YES":"NO";} template<typename type> vector<vector<type>> product(vector<vector<type>> &a, vector<vector<type>> &b){ vector<vector<type>> res(a.size(),vector<type>(b[0].size(),0)); for(int i = 0; i < a.size(); i++){ for(int j = 0; j < b[0].size(); j++){ for(int k = 0; k < b.size(); k++){ res[i][j] += a[i][k]*b[k][j]; res[i][j] %= MOD; } } } return res; } template<typename type> vector<vector<type>> matrix_power(vector<vector<type>> matrix, long long n){ vector<vector<type>> res(matrix.size(), vector<type>(matrix.size(),0)); for(int i = 0; i < matrix.size(); i++) res[i][i] = 1; for( ; n > 0; n >>= 1){ if(n&1) res = product(res,matrix); matrix = product(matrix,matrix); } return res; } signed main(){ cin.tie(0); ios::sync_with_stdio(false); cout<<fixed<<setprecision(10); vector<vector<int>> matrix; int N, K; int ans = 0; cin>>N>>K; matrix.resize(K*K*K,vector<int>(K*K*K,0)); for(int i = 0; i < K; i++){ for(int j = 0; j < K; j++){ for(int k = 0; k < K; k++){ matrix[(i+1)%K*K*K + j*K + k][i*K*K + j*K + k]++; matrix[i*K*K + (i+j)%K*K + k][i*K*K + j*K + k]++; matrix[i*K*K + j*K + (j+k)%K][i*K*K + j*K + k]++; } } } matrix = matrix_power(matrix, N); for(int i = 0; i < K; i++){ for(int j = 0; j < K; j++){ ans += matrix[i*K*K + j*K][0]; ans %= MOD; } } cout<<ans<<endl; return 0; }