結果
問題 | No.1810 RGB Biscuits |
ユーザー |
![]() |
提出日時 | 2023-05-25 01:26:34 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 16 ms / 2,000 ms |
コード長 | 2,603 bytes |
コンパイル時間 | 1,158 ms |
コンパイル使用メモリ | 112,884 KB |
最終ジャッジ日時 | 2025-02-13 04:49:36 |
ジャッジサーバーID (参考情報) |
judge2 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 20 |
ソースコード
#include <iostream> #include <vector> #include <cmath> #include <map> #include <set> #include <iomanip> #include <queue> #include <algorithm> #include <numeric> #include <deque> #include <complex> #include <cassert> using namespace std; using ll = long long; const ll modc = 1e9+7; class mint { ll x; public: mint(ll x=0) : x((x%modc+modc)%modc) {} mint operator-() const { return mint(-x); } mint& operator+=(const mint& a) { if ((x += a.x) >= modc) x -= modc; return *this; } mint& operator-=(const mint& a) { if ((x += modc-a.x) >= modc) x -= modc; return *this; } mint& operator*=(const mint& a) { (x *= a.x) %= modc; return *this; } mint operator+(const mint& a) const { mint res(*this); return res+=a; } mint operator-(const mint& a) const { mint res(*this); return res-=a; } mint operator*(const mint& a) const { mint res(*this); return res*=a; } mint pow(ll t) const { if (!t) return 1; mint a = pow(t>>1); a *= a; if (t&1) a *= *this; return a; } mint inv() const { return pow(modc-2); } mint& operator/=(const mint& a) { return (*this) *= a.inv(); } mint operator/(const mint& a) const { mint res(*this); return res/=a; } bool operator == (const mint& a) const{ return x == a.x; } friend ostream& operator<<(ostream& os, const mint& m){ os << m.x; return os; } friend istream& operator>>(istream& ip, mint&m) { ip >> m.x; return ip; } }; vector<vector<mint>> dot(vector<vector<mint>> &a, vector<vector<mint>> &b){ vector<vector<mint>> c(2, vector<mint>(2)); for (int i=0; i<2; i++){ for (int k=0; k<2; k++){ for (int j=0; j<2; j++) c[i][j] += (a[i][k] * b[k][j]); } } return c; } //A^N vector<vector<mint>> pow(vector<vector<mint>> A, ll N){ vector<vector<mint>> B = {{1, 0}, {0, 1}}; while(N){ if (N % 2 == 1) B = dot(B, A); N >>= 1; A = dot(A, A); } return B; } int main(){ ll N, A, B, T; mint r, g, b; cin >> A >> B >> N; vector<vector<mint>> mat = {{A, B}, {1, 0}}; while(N){ N--; cin >> T; vector<vector<mint>> mat2 = pow(mat, T/2); r = mat2[0][0] + mat2[0][1]; g = mat2[1][0] + mat2[1][1]; b = 0; if (T % 2 == 1) b += r*A+g*B; cout << r+g+b << endl; } return 0; }