結果
問題 | No.1810 RGB Biscuits |
ユーザー |
![]() |
提出日時 | 2022-01-14 21:41:02 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 28 ms / 2,000 ms |
コード長 | 2,188 bytes |
コンパイル時間 | 4,820 ms |
コンパイル使用メモリ | 264,940 KB |
最終ジャッジ日時 | 2025-01-27 11:09:45 |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 20 |
ソースコード
#include <stdio.h> #include <bits/stdc++.h> #include <atcoder/all> using namespace atcoder; using mint = modint1000000007; using namespace std; #define rep(i,n) for(int i=0;i<(n);i++) #define Inf 1000000001 template <class S, S (*op0)(S, S), S (*e0)(), S (*op1)(S, S), S (*e1)() > struct matrix{ vector<vector<S>> v; int _h,_w; matrix(vector<vector<S>> X){ v = X; _h = X.size(); _w = 0; if(X.size()>0)_w = X[0].size(); } matrix(int h,int w){ v.resize(h,vector<S>(w,e0())); _h = h; _w = w; } void add_element(int from,int to,S x){ v[to][from] = op0(v[to][from],x); } matrix e(){ assert(_h==_w); matrix<S,op0,e0,op1,e1> ret(_h,_w); for(int i=0;i<_h;i++){ for(int j=0;j<_w;j++){ if(i==j)ret.v[i][j] = e1(); else ret.v[i][j] = e0(); } } return ret; } matrix &operator*=(const matrix &another){ matrix<S,op0,e0,op1,e1> ret(_h,another._w); for(int i=0;i<_h;i++){ for(int j=0;j<another._w;j++){ ret.v[i][j] = e0(); for(int k=0;k<_w;k++){ ret.v[i][j] = op0(ret.v[i][j],op1(v[i][k],another.v[k][j])); } } } v = ret.v; return (*this); } matrix operator*(const matrix &another)const{ return (matrix(*this)*=another); } matrix pow(long long cnt){ matrix<S,op0,e0,op1,e1> ret = e(); auto temp = *this; while(cnt!=0LL){ if((cnt&1)==1){ ret *= temp; } temp *= temp; cnt>>=1; } return ret; } }; mint op0(mint a,mint b){ return a+b; } mint e0(){ return 0; } mint op1(mint a,mint b){ return a*b; } mint e1(){ return 1; } int main(){ int A,B; cin>>A>>B; matrix<mint,op0,e0,op1,e1> X(3,3); matrix<mint,op0,e0,op1,e1> Y(3,3); X.add_element(0,0,1); X.add_element(1,1,1); X.add_element(2,2,1); X.add_element(0,2,A); X.add_element(1,2,B); Y.add_element(0,1,1); Y.add_element(2,0,1); auto Z = Y*X; int _t; cin>>_t; rep(_,_t){ long long n; cin>>n; auto temp = Z.pow(n/2); if(n%2==1)temp = X*temp; matrix<mint,op0,e0,op1,e1> t(3,1); t.v[0][0] = 1; t.v[1][0] = 1; t = temp*t; mint ans = 0; rep(i,3){ ans += t.v[i][0]; } cout<<ans.val()<<endl; } return 0; }