【蓝桥杯】试题 算法提高 递推求值

问题描述

资源限制
　　时间限制：1.0s 内存限制：256.0MB
问题描述
　　已知递推公式：
　　F(n, 1)=F(n-1, 2) + 2F(n-3, 1) + 5,
　　F(n, 2)=F(n-1, 1) + 3F(n-3, 1) + 2F(n-3, 2) + 3.
　　初始值为：F(1, 1)=2, F(1, 2)=3, F(2, 1)=1, F(2, 2)=4, F(3, 1)=6, F(3, 2)=5。
　　输入n，输出F(n, 1)和F(n, 2)，由于答案可能很大，你只需要输出答案除以99999999的余数。
　　
输入格式　　
　　输入第一行包含一个整数n。
输出格式
　　输出两行，第一行为F(n, 1)除以99999999的余数，第二行为F(n, 2)除以99999999的余数。

输入样例

4

输出样例

14
21

解题思路

这道题目应该构造一个矩阵A，可以满足 x(n-1)* A=x(n)，具体如何构造详见矩阵构造方法。这类构造还需要多加练习。
　　通过学习可以得到构造出这样的式子：
[f(n-1,1) f(n-1,2) f(n-2,1) f(n-2,2) f(n-3,1) f(n-3,2) 5 3]*A=[f(n,1) f(n,2) f(n-1,1) f(n-1,2) f(n-2,1) f(n-2,2) 5 3]
计算后得到矩阵A为：
0 1 1 0 0 0 0 0
1 0 0 1 0 0 0 0
0 0 0 0 1 0 0 0
0 0 0 0 0 1 0 0
2 3 0 0 0 0 0 0
0 2 0 0 0 0 0 0
1 0 0 0 0 0 1 0
0 1 0 0 0 0 0 1
　　可得到最后的答案:[f(n,1) f(n,2) 6 5 1 4 5 3]=[6 5 1 4 2 3 5 3] * A^n-3，因此只需计算前两个值即可。
　　在进行求解过程中需要计算矩阵A的N次幂，为了用尽量少的时间计算出答案，使用了快速幂计算，方法如下：
　　例如计算2¹⁹=2¹⁶ * 2² * 2¹
　　计算方法：19的二进制位10011，1所对应的就是因数的次方。res计算此时的“1”对应的次方，当为1时，乘于答案，当不为1时，只需要继续计算次方大小。

#include<iostream>
using namespace std;
int main()
{
	int x=2,n=19;
	int ans=1,res=x;
	while(n)
	{
		//19（10011）与00001做与运算 
		//判断19的二进制中最后一位是几
		//当为1时，ans=res*ans;res=res*res;
		//当不为1时，只完成res=res*res; 
		if(n&1)	
			ans=ans*res;
		res=res*res;
		//n右移一格
		n=n>>1; 
	}
	cout<<ans;
	return 0;
 } 

矩阵乘法计算较简单，采用模拟的方法计算即可：

//矩阵相乘：矩阵A为a*b,矩阵B为b*c; 
mat mulit(mat A,mat B,int a,int b,int c)
{
	mat C;
	memset(C.a,0,sizeof(C.a));
	for(int i=0;i<a;i++)
	        for(int j=0;j<c;j++)
	                for(int k=0;k<b;k++)
	                        C.a[i][j]=(C.a[i][j]+(A.a[i][k]*B.a[k][j])%99999999)%99999999;
	return C;               
}

解题代码

#include<iostream>
#include<cstring>
using namespace std;
//存储矩阵A 
struct mat
{
	long long a[8][8];
};
//矩阵相乘矩阵A为a*b,矩阵B为b*c; 
mat mulit(mat A,mat B,int a,int b,int c)
{
	mat C;
	memset(C.a,0,sizeof(C.a));
	for(int i=0;i<a;i++)
	        for(int j=0;j<c;j++)
	                for(int k=0;k<b;k++)
	                        C.a[i][j]=(C.a[i][j]+(A.a[i][k]*B.a[k][j])%99999999)%99999999;
	return C;
	               
}
int main()
{
	mat A={ 0,1,1,0,0,0,0,0,
                1,0,0,1,0,0,0,0,
                0,0,0,0,1,0,0,0,
                0,0,0,0,0,1,0,0,
                2,3,0,0,0,0,0,0,
                0,2,0,0,0,0,0,0,
                1,0,0,0,0,0,1,0,
                0,1,0,0,0,0,0,1};
        mat E;
        memset(E.a,0,sizeof(E.a));
        for(int i=0;i<8;i++)
           E.a[i][i]=1;
        long long s[8]={6,5,1,4,2,3,5,3};
        //注意无论输入的n或者计算的sum 都是long long 型 
        long long n;
        cin>>n;
        if(n==1) cout<<"2"<<endl<<"3"<<endl;
	    if(n==2) cout<<"1"<<endl<<"4"<<endl;
	    if(n==3) cout<<"6"<<endl<<"5"<<endl;
	    if(n>=4){
		    n=n-3;
		    while(n)
		    {
			    if(n&1)
			         E=mulit(E,A,8,8,8);
			    A=mulit(A,A,8,8,8);
			    n>>=1;
		    }//得到A的幂方
		    long long sum1=0,sum2=0;
		    for(int i=0;i<8;i++)
		    {
			    sum1=(sum1+(s[i]*E.a[i][0])%99999999)%99999999;
			    sum2=(sum2+(s[i]*E.a[i][1])%99999999)%99999999;
		     } 
		     cout<<sum1<<endl<<sum2<<endl;
	    }	
	    return 0;
 }