結果
問題 | No.2156 ぞい文字列 |
ユーザー |
![]() |
提出日時 | 2023-07-07 08:21:04 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 2 ms / 2,000 ms |
コード長 | 1,871 bytes |
コンパイル時間 | 4,037 ms |
コンパイル使用メモリ | 257,544 KB |
最終ジャッジ日時 | 2025-02-15 06:32:33 |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 16 |
ソースコード
#include <bits/stdc++.h> using namespace std; #include <atcoder/all> using namespace atcoder; using ll = long long; #define rep(i, n) for (ll i = 0; i < ll(n); i++) using mint = modint998244353; // https://youtu.be/ylWYSurx10A?t=2352 template<typename T> struct Matrix { int h, w; vector<vector<T>> d; Matrix() {} Matrix(int h, int w, T val=0): h(h), w(w), d(h, vector<T>(w,val)) {} Matrix& unit() { assert(h == w); rep(i,h) d[i][i] = 1; return *this; } const vector<T>& operator[](int i) const { return d[i];} vector<T>& operator[](int i) { return d[i];} Matrix operator*(const Matrix& a) const { assert(w == a.h); Matrix r(h, a.w); rep(i,h)rep(k,w)rep(j,a.w) { r[i][j] += d[i][k]*a[k][j]; } return r; } Matrix pow(long long t) const { assert(h == w); if (!t) return Matrix(h,h).unit(); if (t == 1) return *this; Matrix r = pow(t>>1); r = r*r; if (t&1) r = r*(*this); return r; } // https://youtu.be/-j02o6__jgs?t=11273 /* mint only mint det() { assert(h == w); mint res = 1; rep(k,h) { for (int i = k; i < h; ++i) { if (d[i][k] == 0) continue; if (i != k) { swap(d[i],d[k]); res = -res; } } if (d[k][k] == 0) return 0; res *= d[k][k]; mint inv = mint(1)/d[k][k]; rep(j,h) d[k][j] *= inv; for (int i = k+1; i < h; ++i) { mint c = d[i][k]; for (int j = k; j < h; ++j) d[i][j] -= d[k][j]*c; } } return res; } //*/ }; int main(){ ll n; cin >> n; Matrix<mint> mat(3,3); mat[0][0] = 1; mat[0][1] = 1; mat[1][0] = 1; mat[1][2] = 1; mat[2][2] = 1; mat = mat.pow(n-2); Matrix<mint> v0(3,1); v0[1][0] = 1; v0[2][0] = 1; Matrix<mint> ans = mat*v0; cout << (ans[0][0]+ans[1][0]).val() << endl; }