CF450B Jzzhu and Sequences（矩阵加速）

CF450B Jzzhu and Sequences

大佬留言：这。这。不就是矩乘的模板吗，切掉它！！

You are given xx and yy , please calculate $f_{n}(mod(10^{9}+7))$.

原式：$f_{i}=f_{i-1}+f_{i+1}$

转换一下：$f_{i+1}=f_{i}-f_{i-1}$

相当于$f_{i}=f_{i-1}-f_{i-2}$

有没有发现它跟斐波那契通项公式有点儿类似？

的确是这样的，那么转移矩阵也类似：

$0 -1$

$1 1$

矩阵快速幂好了

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>

#define LL long long
#define N 10
using namespace std;

class Martix{
public:
    LL n,m;
    LL A[N][N];
    Martix(){
        memset(A,,sizeof(A));
    }
};
const LL mod=;

Martix operator * (Martix A,Martix B){
    Martix C;
    int n=A.n,m=B.m,p=A.m;
    C.n=n,C.m=m;
    for(LL i=;i<=n;i++)
        for(LL j=;j<=m;j++)
            for(LL k=;k<=p;k++)
                C.A[i][j]=((A.A[i][k]*B.A[k][j]+mod)%mod+C.A[i][j]%mod+mod)%mod;
    return C;
}

LL x,y,n;

Martix A,B;
inline LL pow(){
    for(;n;n>>=,A=A*A)
        if(n&) B=B*A;
    return B.A[][];
}

int main()
{
    scanf("%lld%lld%lld",&x,&y,&n);

    A.n=A.m=;
    A.A[][]=,A.A[][]=-,A.A[][]=,A.A[][]=;
    B.n=,B.m=;
    B.A[][]=(x+mod)%mod,B.A[][]=(y+mod)%mod;
    if(n==) printf("%lld",(x+mod)%mod);
    else if(n==) printf("%lld",(y+mod)%mod);
    else n-=,printf("%lld\n",pow());
    return ;
}