算法提高 递推求值
时间限制: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
数据规模和约定
1<=n<=10^18。
ps: 矩阵快速幂的应用。第一次遇到要将两个递推公式合并。根据题目给的两个递推公式,我们可以得到一个矩阵向量,先记为Fn, 如图:
这题与之前推导矩阵递推公式不同,需要对原来的递推公式稍加变形,如增加f(n - 2).
接下来求出矩阵Fn-1, 然后求出递推的矩阵(相当于等比数列的公比Fn-1 * q = Fn)和F1 = {6,5,1,4,2,3,5,3},如图
注意:求解过程需要注意模运算的正确性。
#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#define mod 99999999
using namespace std;
const int N = 8;
void multi(long long matrix1[][N], long long matrix2[][N])
{
long long temp[N][N]= {0};
for(int i = 0; i < N; i++)
for(int j = 0; j < N; j++)
for(int k = 0; k < N; k++)
{
temp[i][k] += ((matrix1[i][j] % mod) * (matrix2[j][k] % mod)) % mod;
temp[i][k] = temp[i][k] % mod;
}
for(int i = 0; i < N; i++)
for(int j = 0; j < N; j++)
matrix1[i][j] = temp[i][j];
}
long long ans[N][N];
void pow(long long matrix[][N], long long n)
{
memset(ans, 0, sizeof ans);
for(int i = 0; i < N; i++)ans[i][i] = 1;
while(n > 0)
{
if(n & 1) multi(ans, matrix);
multi(matrix, matrix);
n = n >> 1;
}
}
void multi1(long long matrix1[][8], long long matrix2[][1])
{
long long temp[N][N] = {0};
for(int i = 0; i < N; i++)
for(int j = 0; j < N; j++)
for(int k = 0; k < 1; k++)
{
temp[i][k] += ((matrix1[i][j] % mod) * (matrix2[j][k] % mod)) % mod;
temp[i][k] = temp[i][k] % mod;
}
for(int i = 0; i < N; i++)
for(int j = 0; j < 1; j++)
{
matrix2[i][j] = temp[i][j];
}
}
int main()
{
long long n;
while(cin>>n)
{
long long A[8][1] = {6,5,1,4,2,3,5,3};
long long matrix[8][8] = {
0,1,0,0,2,0,1,0,
1,0,0,0,3,2,0,1,
1,0,0,0,0,0,0,0,
0,1,0,0,0,0,0,0,
0,0,1,0,0,0,0,0,
0,0,0,1,0,0,0,0,
0,0,0,0,0,0,1,0,
0,0,0,0,0,0,0,1 };
pow(matrix, n - 1);
if(n == 1){
memset(ans, 0, sizeof ans);
for(int i = 0; i < 8; i++)ans[i][i] = 1;
}
multi1(ans, A);
cout<<A[4][0]<<endl;
cout<<A[5][0]<<endl;
}
return 0;
}