結果
問題 | No.658 テトラナッチ数列 Hard |
ユーザー |
![]() |
提出日時 | 2020-07-10 23:02:32 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 479 ms / 2,000 ms |
コード長 | 1,954 bytes |
コンパイル時間 | 1,087 ms |
コンパイル使用メモリ | 84,428 KB |
最終ジャッジ日時 | 2025-01-11 19:09:52 |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 8 |
ソースコード
#include <iostream> #include <vector> using namespace std; using ll = long long; #define rep(i,n) for(int i=0,_i=(n);i<_i;++i) template<int MOD> struct MInt { long long val; constexpr MInt(long long val = 0) : val(val % MOD) { if (val < 0) val += MOD; } MInt operator+(const MInt& n) const { return MInt(val) += n; } MInt operator*(const MInt& n) const { return MInt(val) *= n; } MInt& operator+=(const MInt& n) { val = (val + n.val) % MOD; return *this; } MInt& operator*=(const MInt& n) { val = (val * n.val) % MOD; return *this; } friend ostream& operator<<(ostream& os, const MInt& n) { os<<n.val; return os; } }; using mint = MInt<17>; template<class T> struct Matrix { int width, height; vector<vector<T>> v; Matrix(int w) : width(w), height(w), v(w, vector<T>(w, 0)) {}; Matrix(int w, int h) : width(w), height(h), v(w, vector<T>(h, 0)) {}; Matrix(vector<vector<T>> _v) : width(_v.size()), height(_v[0].size()), v(_v) {}; Matrix operator*(const Matrix& n) const { Matrix ans = v; return ans *= n; } Matrix& operator*=(const Matrix& rhs) { Matrix<T> ans(height, rhs.width); for (int i = 0; i < height; ++i) for (int j = 0; j < rhs.width; ++j) for (int k = 0; k < width; ++k) ans[i][j] += v[i][k] * rhs.v[k][j]; width = rhs.width; v = ans.v; return *this; } Matrix pow(ll N) { Matrix pow = *this, ans(width); for (int i = 0; i < width; ++i) ans[i][i] = 1; for (ll p = N; p > 0; p >>= 1, pow *= pow) if (p & 1) ans *= pow; return ans; } vector<T>& operator[](int i) { return v[i]; } }; int main() { int N; cin >> N; Matrix<mint> m(4); m.v[0][0] = m.v[0][1] = m.v[0][2] = m.v[0][3] = 1; m.v[1][0] = m.v[2][1] = m.v[3][2] = 1; rep(i, N) { ll a; cin >> a; if (a <= 4) { cout << a / 4 << endl; continue; } Matrix<mint> p(4, 1); p[0][0] = 1; cout << (m.pow(a-4) * p)[0][0] << endl; } return 0; }