递推数列（矩阵快速幂）

题目描述

给定a0,a1,以及an=p*a(n-1) + q*a(n-2)中的p,q。这里n >= 2。 求第k个数对10000的模。

输入描述:

输入包括5个整数：a0、a1、p、q、k。

输出描述:

第k个数a(k)对10000的模。

示例1

输入

20 1 1 14 5

输出

8359

思路:矩阵快速幂

#include<iostream>
#include<string.h> 
using namespace std;
long long  n,N=2;
long long a0,a1,p,q,k; 
struct matrix//用结构体封装一个二维数组！！！ 
{
    long long a[100][100];
}origin,res;//res.a存储结果 ，origin.a是需要运算的矩阵 ，n是次幂数 ,N是矩阵的行列数 
matrix multiply(matrix x,matrix y)//矩阵相乘函数 
{
       matrix temp;
       memset(temp.a,0,sizeof(temp.a));
       for(int i=0;i<N;i++)
       {
               for(int j=0;j<N;j++)
               {
                       for(int k=0;k<N;k++)
                       {
                               temp.a[i][j]+=x.a[i][k]*y.a[k][j];
                       }
                        temp.a[i][j]%=10000;
               }
       }
       return temp;
}
void calc(int n)
{
     memset(res.a,0,sizeof(res.a)); 
     for(int i=0;i<N;i++)
     	res.a[i][i]=1;                  //将res.a初始化为单位矩阵 
     while(n)
     {
         if(n&1)
             res=multiply(res,origin);
         n>>=1;
         origin=multiply(origin,origin);
     }
}
int main()
{
	long long ans;
	while(cin>>a0>>a1>>p>>q>>k)
	{
		if(k==1)
			cout<<a1<<endl;
		else if(k==0)
			cout<<a0<<endl;
		else
		{
			origin.a[0][0]=p%10000;
			origin.a[0][1]=q%10000;
			origin.a[1][0]=1;	
			origin.a[1][1]=0;
			calc(k-1);
			ans=res.a[0][0]*a1+res.a[0][1]*a0;
			cout<<ans%10000<<endl;
		}
		
	}
	return 0;
}