結果
問題 | No.658 テトラナッチ数列 Hard |
ユーザー |
👑 |
提出日時 | 2022-10-22 02:52:47 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 290 ms / 2,000 ms |
コード長 | 1,939 bytes |
コンパイル時間 | 936 ms |
コンパイル使用メモリ | 82,540 KB |
最終ジャッジ日時 | 2025-02-08 10:54:27 |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 8 |
ソースコード
#include <iostream> #include <vector> using namespace std; template <typename T> struct mat{ vector<vector<T>> x; int h,w; mat():x(vector<vector<T>>()){} mat(int h,int w):x(vector<vector<T>>(h,vector<T>(w))),h(h),w(w){} mat(int h,int w, T c):x(vector<vector<T>>(h,vector<T>(w,c))),h(h),w(w){} mat(vector<vector<T>> A):x(A),h(A.size()),w(A[0].size()){} vector<T>& operator[](int i){return x[i];} mat& operator*=(mat& y){ mat<T> ret(h,y.w,0); if(w != y.h){ for(int i = 0; h > i; i++){ for(int j = 0; y.w > j; j++){ ret[i][j] = -1; } } }else{ for(int i = 0; h > i; i++){ for(int j = 0; y.w > j; j++){ for(int k = 0; w > k; k++){ ret[i][j] = (ret[i][j] + x[i][k]*y[k][j])%17; } } } } for(int i = 0; h > i; i++){ x[i].resize(y.w); } w = y.w; for(int i = 0; h > i; i++){ for(int j = 0; y.w > j; j++){ x[i][j] = ret[i][j]; } } return *this; } mat operator*(mat& y){return mat(*this) *= y;} mat pow(long long n){//正方行列のみ mat<T> res(h,w); mat<T> ret(h,w,0); mat<T> a(h,w); for(int i = 0; h > i; i++){ ret[i][i] = 1; } for(int i = 0; h > i; i++){ for(int j = 0; w > j; j++){ a[i][j] = (*this)[i][j]; } } while(n > 0){ if(n & 1){ ret *= a; } a *= a; n/=2; } for(int i = 0; h > i; i++){ for(int j = 0; w > j; j++){ res[i][j] = ret[i][j]; } } return res; } }; int main(){ int q;cin>>q; vector<vector<long long>> A{{1,1,1,1},{1,0,0,0},{0,1,0,0},{0,0,1,0}}; vector<vector<long long>> B{{1},{0},{0},{0}}; mat<long long> a(A),b(B); for(int i = 0; q > i; i++){ long long n;cin>>n; if(n < 4){ cout << 0 << endl; continue; } auto c = a.pow(n-4)*b; cout << c[0][0]%17 << endl; } }