結果

問題 No.1810 RGB Biscuits
ユーザー ringo99ringo99
提出日時 2022-01-15 15:05:26
言語 C++14
(gcc 11.2.0 + boost 1.78.0)
結果
AC  
実行時間 28 ms / 2,000 ms
コード長 1,610 Byte
コンパイル時間 1,675 ms
使用メモリ 3,560 KB
最終ジャッジ日時 2022-06-28 21:43:07
合計ジャッジ時間 2,997 ms
ジャッジサーバーID
(参考情報)
judge15 / judge14
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
testcase_00 AC 1 ms
3,476 KB
testcase_01 AC 2 ms
3,476 KB
testcase_02 AC 1 ms
3,528 KB
testcase_03 AC 2 ms
3,412 KB
testcase_04 AC 2 ms
3,512 KB
testcase_05 AC 25 ms
3,408 KB
testcase_06 AC 28 ms
3,412 KB
testcase_07 AC 27 ms
3,480 KB
testcase_08 AC 27 ms
3,404 KB
testcase_09 AC 27 ms
3,504 KB
testcase_10 AC 18 ms
3,400 KB
testcase_11 AC 3 ms
3,560 KB
testcase_12 AC 3 ms
3,480 KB
testcase_13 AC 2 ms
3,476 KB
testcase_14 AC 6 ms
3,472 KB
testcase_15 AC 2 ms
3,408 KB
testcase_16 AC 2 ms
3,560 KB
testcase_17 AC 14 ms
3,500 KB
testcase_18 AC 14 ms
3,472 KB
testcase_19 AC 2 ms
3,408 KB
testcase_20 AC 1 ms
3,512 KB
testcase_21 AC 4 ms
3,552 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std ;
typedef long long ll ;
typedef long double ld ;
typedef pair<ll,ll> P ;
typedef tuple<bool,ll,ll,ll> TP ;
#define chmin(a,b) a = min(a,b)
#define chmax(a,b) a = max(a,b)
#define bit_count(x) __builtin_popcountll(x)
#define gcd(a,b) __gcd(a,b)
#define lcm(a,b) a / gcd(a,b) * b
#define rep(i,n) for(int i = 0 ; i < n ; i++)
#define rrep(i,a,b) for(int i = a ; i < b ; i++)
#define endl "\n"

const int mod = 1000000007 ;

vector<vector<ll>> A(55,vector<ll>(55)) ;

struct MATRIX{
    vector<vector<ll>> mat ;
    MATRIX(vector<vector<ll>> X){ mat = X ; }
    // A * B
    vector<vector<ll>> calculation(vector<vector<ll>> X , vector<vector<ll>> Y){
        int x = X.size() , y = Y[0].size() , n = Y.size() ;
        vector<vector<ll>> res(x,vector<ll>(y)) ;
        rep(i,x) rep(j,y) rep(k,n) {
            (res[i][j] += X[i][k] * Y[k][j]) %= mod ;
        }
        return res ;
    }

    void dynamic(vector<vector<ll>> A , ll k){
        int n = A.size() ;
        while(k > 0){
            if(k & 1) mat = calculation(A,mat) ;
            A = calculation(A,A) ;
            k >>= 1 ;
        }
    }

    inline ll operator [] (int i) { return mat[i][0] ; }
};

ll a , b , n ;

int main(){
    cin >> a >> b >> n ;
    vector<vector<ll>> A = {{a,b},{1,0}} , X = {{1},{1}} ;
    rep(i,n){
        ll t ;
        cin >> t ;
        ll s = t / 2 ;
        MATRIX mat(X) ;
        mat.dynamic(A,s) ;
        ll v = t % 2 == 0 ? 0 : (mat[0] * a % mod + mat[1] * b % mod) % mod ;
        ll res = (mat[0] + mat[1] + v) % mod ;
        cout << res << endl ;
    }
}
0