結果
問題 | No.1879 How many matchings? |
ユーザー |
![]() |
提出日時 | 2022-03-18 22:57:53 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 2 ms / 2,000 ms |
コード長 | 2,871 bytes |
コンパイル時間 | 2,536 ms |
コンパイル使用メモリ | 217,824 KB |
最終ジャッジ日時 | 2025-01-28 10:36:20 |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 15 |
ソースコード
#include<bits/stdc++.h>namespace {#pragma GCC diagnostic push#pragma GCC diagnostic ignored "-Wunused-function"#include<atcoder/all>#pragma GCC diagnostic popusing namespace std;using namespace atcoder;#define rep(i,n) for(int i = 0; i < (int)(n); i++)#define rrep(i,n) for(int i = (int)(n) - 1; i >= 0; i--)#define all(x) begin(x), end(x)#define rall(x) rbegin(x), rend(x)template<class T> bool chmax(T& a, const T& b) { if (a < b) { a = b; return true; } else return false; }template<class T> bool chmin(T& a, const T& b) { if (b < a) { a = b; return true; } else return false; }using ll = long long;using P = pair<int,int>;using VI = vector<int>;using VVI = vector<VI>;using VL = vector<ll>;using VVL = vector<VL>;using mint = modint1000000007;template<class T, T (*add)(T, T), T (*zero)(), T (*mul)(T, T), T (*one)(), int N>struct Mat : array<array<T, N>, N> {using M = Mat<T, add, zero, mul, one, N>;void make_identity() {for (int i = 0; i < N; i++) {for (int j = 0; j < N; j++) {(*this)[i][j] = zero();}}for (int i = 0; i < N; i++) {(*this)[i][i] = one();}}M& operator+=(const M& rhs) {for (int i = 0; i < N; i++) {for (int j = 0; j < N; j++) {(*this)[i][j] = add((*this)[i][j], rhs[i][j]);}}return *this;}M& operator*=(const M& rhs) {static M temp;for (int i = 0; i < N; i++) {for (int j = 0; j < N; j++) {temp[i][j] = zero();}}for (int i = 0; i < N; i++) {for (int j = 0; j < N; j++) {for (int k = 0; k < N; k++) {temp[i][k] = add(temp[i][k], mul((*this)[i][j], rhs[j][k]));}}}*this = temp;return *this;}template <class I>void inplace_pow(I k) {assert(k >= 0);static M temp;temp = *this;make_identity();while (k) {if (k & 1) *this *= temp;k >>= 1;if (k) temp *= temp;}}friend ostream& operator<<(ostream& os, const M& A) {for (int i = 0; i < N; i++) {for (int j = 0; j < N; j++) {os << A[i][j] << " \n"[j + 1 == N];}}return os;}};mint add(mint x, mint y) { return x + y; }mint zero() { return mint(); }mint mul(mint x, mint y) { return x * y; }mint one() { return mint::raw(1); }using M = Mat<mint, add, zero, mul, one, 8>;} int main() {ios::sync_with_stdio(false);cin.tie(0);M a;rep(b, 2) rep(s, 1 << 2) {if (s & 1) a[b << 2 | s][b << 2 | s >> 1] = 1;else {if (b == 0) a[b << 2 | s][1 << 2 | s >> 1] = 1;if (s == 0) a[b << 2 | s][b << 2 | 1] = 1;a[b << 2 | s][b << 2 | s >> 1 | 2] = 1;}}ll n;cin >> n;if (n == 1) {cout << 1 << '\n';return 0;}a.inplace_pow(n);mint ans;rep(b, 2) ans += a[0][b << 2 | 0];cout << ans.val() << '\n';}