結果

問題 No.194 フィボナッチ数列の理解(1)
ユーザー yaoshimax
提出日時 2015-05-15 00:59:19
言語 PyPy2
(7.3.15)
結果
AC  
実行時間 139 ms / 5,000 ms
コード長 1,843 bytes
コンパイル時間 1,363 ms
コンパイル使用メモリ 76,580 KB
実行使用メモリ 79,368 KB
最終ジャッジ日時 2024-07-06 03:46:39
合計ジャッジ時間 6,880 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

diff # 

N,K=map(int,raw_input().split())
A=map(int,raw_input().split())
mod=1000000007
s=sum(A)
S=sum(A)
if N>30:
    for i in range(N,K):
        s-=A[i%N]
        A[i%N]=s+A[i%N]
        A[i%N]%=mod
        s+=A[i%N]
        S+=A[i%N]
        S%=mod
        s%=mod
    print A[(K-1)%N],S
else:
    mat=[[0 for i in range(N+1)] for j in range(N+1)]
    for i in range(N):
        mat[i][i+1]=1
    mat[N][N]=2
    mat[N][0]-=1
    vec=[0 for i in range(N+1)]
    vec[0]=0
    for i in range(1,N+1):
        vec[i]=A[i-1]+vec[i-1]
    m=[[0 for i in range(N+1)] for j in range(N+1)]
    for i in range(N+1):
        m[i][i]=1
    K-=N
    while K>0:
    #    for i in range(N+1):
    #        for j in range(N+1):
    #            print m[i][j],
    #        print "   ",
    #        for j in range(N+1):
    #            print mat[i][j],
    #        print "   ",
    #        print vec[i]
    #    print
        if K%2==1:
            m2=[[0 for i in range(N+1)] for j in range(N+1)]
            for i in range(N+1):
                for j in range(N+1):
                    for k in range(N+1):
                        m2[i][j]+=mat[i][k]*m[k][j]
                        m2[i][j]%=mod
            m = m2
        mat2=[[0 for i in range(N+1)] for j in range(N+1)]
        for i in range(N+1):
            for j in range(N+1):
                for k in range(N+1):
                    mat2[i][j]+=mat[i][k]*mat[k][j]
                    mat2[i][j]%=mod
        mat=mat2
        K/=2
    ans =0
    ans2 =0
    #for i in range(N+1):
    #    for j in range(N+1):
    #        print m[i][j],
    #    print "   ",
    #    for j in range(N+1):
    #        print mat[i][j],
    #    print
    #    print
    for i in range(N+1):
        ans+=m[N][i]*vec[i]
        ans2+=m[N-1][i]*vec[i]
        ans%=mod
        ans2%=mod
    print (ans-ans2)%mod, ans
0