結果

問題 No.194 フィボナッチ数列の理解(1)
ユーザー btkbtk
提出日時 2017-05-20 17:49:44
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 18 ms / 5,000 ms
コード長 3,072 bytes
コンパイル時間 1,761 ms
コンパイル使用メモリ 175,488 KB
実行使用メモリ 19,200 KB
最終ジャッジ日時 2024-09-19 00:28:53
合計ジャッジ時間 3,226 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 8 ms
5,376 KB
testcase_03 AC 8 ms
5,376 KB
testcase_04 AC 8 ms
5,376 KB
testcase_05 AC 8 ms
5,376 KB
testcase_06 AC 8 ms
5,376 KB
testcase_07 AC 7 ms
5,376 KB
testcase_08 AC 8 ms
5,376 KB
testcase_09 AC 8 ms
5,376 KB
testcase_10 AC 7 ms
5,376 KB
testcase_11 AC 7 ms
5,376 KB
testcase_12 AC 8 ms
5,376 KB
testcase_13 AC 8 ms
5,376 KB
testcase_14 AC 8 ms
5,376 KB
testcase_15 AC 8 ms
5,376 KB
testcase_16 AC 8 ms
5,376 KB
testcase_17 AC 8 ms
5,376 KB
testcase_18 AC 8 ms
5,376 KB
testcase_19 AC 8 ms
5,376 KB
testcase_20 AC 18 ms
19,072 KB
testcase_21 AC 17 ms
19,072 KB
testcase_22 AC 16 ms
19,200 KB
testcase_23 AC 3 ms
5,376 KB
testcase_24 AC 11 ms
11,008 KB
testcase_25 AC 10 ms
10,112 KB
testcase_26 AC 9 ms
10,112 KB
testcase_27 AC 11 ms
11,904 KB
testcase_28 AC 4 ms
5,376 KB
testcase_29 AC 15 ms
17,792 KB
testcase_30 AC 8 ms
5,376 KB
testcase_31 AC 8 ms
5,376 KB
testcase_32 AC 8 ms
5,376 KB
testcase_33 AC 8 ms
5,376 KB
testcase_34 AC 8 ms
5,376 KB
testcase_35 AC 8 ms
5,376 KB
testcase_36 AC 8 ms
5,376 KB
testcase_37 AC 7 ms
5,376 KB
testcase_38 AC 8 ms
5,376 KB
testcase_39 AC 9 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;

#define CIN_ONLY if(1)
struct cww{cww(){
    CIN_ONLY{
        ios::sync_with_stdio(false);cin.tie(0);
    }
}}star;
#define fin "\n"
#define FOR(i,bg,ed) for(int i=(bg);i<(ed);i++)
#define REP(i,n) FOR(i,0,n)
#define ALL(v) (v).begin(),(v).end()
#define fi first
#define se second
#define pb push_back
#define DEBUG if(0)
#define REC(ret, ...) std::function<ret (__VA_ARGS__)>
template <typename T>inline bool chmin(T &l,T r)
{bool a=l>r;if(a)l=r;return a;}
template <typename T>inline bool chmax(T &l,T r)
{bool a=l<r;if(a)l=r;return a;}
template <typename T>
istream& operator>>(istream &is,vector<T> &v){
    for(auto &it:v)is>>it;
    return is;
}


const int mod=1e9+7;
typedef LL D;
#define REP4(i,n) for(int i=0;i<n;i+=4)
#define SZ 32
int sz=SZ;
struct M{
    D v[SZ][SZ];
    D& operator()(int i,int j){
        return v[i][j];
    }
}mat[64];

void mat_mul(M &a,M& b,M &c){
    REP(i,sz)REP(j,sz)c(i,j)=0;

    D sum[4],tmp;
    REP(k,sz){
        REP4(i,sz){
            sum[0]=a(i,k);
            sum[1]=a(i+1,k);
            sum[2]=a(i+2,k);
            sum[3]=a(i+3,k);
            REP(j,sz){
                tmp=b(k,j);
                c(i,j)+=sum[0]*tmp%mod;
                c(i+1,j)+=sum[1]*tmp%mod;
                c(i+2,j)+=sum[2]*tmp%mod;
                c(i+3,j)+=sum[3]*tmp%mod;
            }
        }
    }
    REP(i,sz)REP(j,sz)c(i,j)%=mod;
}

LL N,K;
LL S[1123456];
LL A[1123456];
typedef pair<LL,LL> P;

P case1(){
    FOR(i,N,K){
        A[i] = (mod + S[i] - S[i - N]) % mod;
        S[i+1] = (A[i] + S[i])%mod;
    }
    return P(A[K-1],S[K]);
}
P case2(){
    P res;
    {
        REP(i,N)REP(j,N)mat[0](i,j)=0;
        REP(i,N-1)mat[0](i+1,i)=1;
        REP(i,N)mat[0](0,i)=1;
        REP(i,50)mat_mul(mat[i],mat[i],mat[i+1]);
        LL bit=K-N;
        vector<LL> f(N);
        REP(i,N)f[i]=A[i];
        reverse(ALL(f));
        for(int l=0;bit;l++,bit>>=1){
            if(bit&1){
                vector<LL> g(N);
                REP(i,N)REP(j,N)g[i]+=mat[l](i,j)*f[j]%mod;
                REP(i,N)g[i]%=mod;
                swap(f,g);
            }
        }
        res.fi=f.front();
    }
    {
        REP(i,N+1)REP(j,N+1)mat[0](i,j)=0;
        REP(i,N)mat[0](i+1,i)=1;
        mat[0](0,0)=2;
        mat[0](0,N)=mod-1;
        REP(i,50)mat_mul(mat[i],mat[i],mat[i+1]);
        LL bit=K-N;
        vector<LL> f(N+1);
        REP(i,N+1)f[i]=S[i];
        reverse(ALL(f));
        for(int l=0;bit;l++,bit>>=1){
            if(bit&1){
                vector<LL> g(N+1);
                REP(i,N+1)REP(j,N+1)g[i]+=mat[l](i,j)*f[j]%mod;
                REP(i,N+1)g[i]%=mod;
                swap(f,g);
            }
        }
        res.se=f.front();
    }

    return res;
}
int main(void){
    cin >> N >> K;
    S[0]=0;        
    REP(i,N){
        cin >> A[i];
        S[i+1] = S[i] + A[i];
        S[i] %= mod;
    }
    
    pair<LL, LL> res;
    if (K <= 1000000)res= case1();
    else res = case2();
    cout << res.fi << " " << res.se << endl;
    return 0;
}
0