51 nod 1126 求递推序列的第N项 矩阵快速幂

来源:互联网 发布:2016张北云计算招聘 编辑:程序博客网 时间:2024/06/01 20:34

题目链接:https://www.51nod.com/onlineJudge/questionCode.html#!problemId=1126
题意:有下面这个式子:F1=1,F2=1.F[n]=(A*F[n-1]+B*F[n-2])%7 (n>=3) .
求给定A,B,N,求F[N].
智障一样的找了一个小时:1.矩阵式n-2次方,2,F1=1,不是等于0啊。

#include<cstdio>#include<queue>#include<iostream>#include<vector>#include<map>#include<cstring>#include<string>#include<set>#include<stack>#include<algorithm>#define cle(a) memset(a,0,sizeof(a))#define inf(a) memset(a,0x3f,sizeof(a))#define ll long long#define Rep(i,a,n) for(int i=a;i<=n;i++)using namespace std;const int INF = ( 2e9 ) + 2;const int mod = 7;int c[2][2];void mul(int e[][2],int a[][2],int b[][2]){    int t[2][2];    memset(t,0,sizeof(t));    for(int i=0;i<2;i++)    for(int j=0;j<2;j++)    for(int k=0;k<2;k++)    t[i][j]=(t[i][j]+((a[i][k]*b[k][j])%mod+mod)%mod)%mod;    for(int i=0;i<2;i++)    for(int j=0;j<2;j++)    e[i][j]=t[i][j];    return;}void qPow(int n){    int ret[2][2];    memset(ret,0,sizeof(ret));    for(int i=0;i<2;i++)ret[i][i]=1;    while(n)    {        if(n&1)        mul(ret,ret,c);        mul(c,c,c);        n>>=1;    }    for(int i=0;i<2;i++)    for(int j=0;j<2;j++)    c[i][j]=ret[i][j];}int main(){    int a,b,n;//  freopen("in.txt","r",stdin);    scanf("%d%d%d",&a,&b,&n);    c[0][0]=0;    c[0][1]=1;    c[1][0]=b;    c[1][1]=a;    qPow(n-2);    printf("%d\n",(c[1][1]+c[1][0])%mod);}
阅读全文
0 0
原创粉丝点击