欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

试题 算法提高 递推求值

程序员文章站 2022-06-04 12:23:58
...

问题描述
  已知递推公式:

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
数据规模和约定
  1<=n<=10^18。
思路:矩阵快速幂
矩阵的构造:把已知条件里得元素分解构成矩阵,然后写出矩阵的方程,比如这道题可以如下构造:

试题 算法提高 递推求值
可以求得矩阵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}
然后使用快速幂求解(每道题因为选取的元素不同,最后构造出的矩阵也不相同。)
如何构造矩阵
参考大佬的博客

 #include<stdio.h>
 #include<string.h>
 #include<iostream>
 using namespace std;
 
 typedef long long ll;
 ll mod=99999999;
struct Matrix
{
	ll mat[8][8];
};
       
Matrix fun(Matrix A,Matrix B)//两个矩阵相乘 
{
	Matrix res;
	memset(res.mat,0,sizeof(res.mat));
	for(int i=0;i<8;i++)
	{
		for(int j=0;j<8;j++)
		{
			for(int k=0;k<8;k++)
			{
				res.mat[i][j]=(res.mat[i][j]+A.mat[i][k]*B.mat[k][j])%mod;
			}
		}
	}
	return res;
}

Matrix fun1(ll n)//矩阵快速幂 
{
	Matrix ans,res;
	ans={       0,1,1,0,0,0,0,0,//这里的初始化只能这样写,不能写成ans.mat不然会报错 
                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};
    memset(res.mat,0,sizeof(res.mat));
    for(int i=0;i<8;i++)
		res.mat[i][i]=1;//构造单位矩阵 
    while(n)
	{
		if(n%2==1)
			res=fun(res,ans);
		ans=fun(ans,ans);
		n/=2;
	}
	return res;
}
 
 int main()
 {
    ll arr[8]={6,5,1,4,2,3,5,3};
    ll n;
	scanf("%I64d",&n);
	if(n==1)
	  printf("2\n3");
	else if(n==2) 
	  printf("1\n4");
	else if(n==3)
	  printf("6\n5");
	else
	{
		n-=3;//从n>3才开始使用矩阵
		Matrix kk=fun1(n);
		ll sum1=0,sum2=0;
    	for(int i=0;i<8;i++)
	    {
		   sum1=(sum1+arr[i]*kk.mat[i][0])%mod;
		   sum2=(sum2+arr[i]*kk.mat[i][1])%mod;
	    }
	   printf("%I64d\n%I64d",sum1,sum2);	
	}
   return 0; 
 }