【SSL】P1530 裴波那契数列III

Description

求数列$f[n]=f[n-1]+f[n-2]+1$的第$N$项.$f[1]=1,f[2]=1$.

Input

$n(1＜n＜{\mathrm{２}}^{31}-1)$

Output

第$Ｎ$项的结果 $mod$ $9973$

Sample Input

12345

Sample Output

8932

正题：

我们考虑将$\left[\begin{array}{ccc}{f}_{n-1}& f_{n-2}& 1\end{array}\right]$转换为$\left[\begin{array}{ccc}{f}_n& f_{n-1}& 1\end{array}\right]$
因为$f[n]=f[n-1]+f[n-2]+1$，所以我们很快得出转移矩阵：
$$\left[\begin{array}{ccc}1& 1& 0\\ 1& 0& 0\\ 1& 0& 1\end{array}\right]$$

code:

#include<iostream>
#include<cstdio>
using namespace std;
int n;
int a[10][10];
int ans[10][10];
int p=9973; 
void multi()
{
	long long c[10][10];
	for(long long i=1; i<=3; i++)
		for(long long j=1; j<=3; j++)
			c[i][j]=0;
	for(long long i=1; i<=3; i++)
		for(long long j=1; j<=3; j++)
			for(long long k=1; k<=3; k++)
				c[i][j]=(c[i][j]+ans[i][k]*a[k][j])%p;
	for(long long i=1; i<=3; i++)
		for(long long j=1; j<=3; j++)
			ans[i][j]=c[i][j];
}
void multi1()
{
	long long c[10][10];
	for(long long i=1; i<=3; i++)
		for(long long j=1; j<=3; j++)
			c[i][j]=0;
	for(long long i=1; i<=3; i++)
		for(long long j=1; j<=3; j++)
			for(long long k=1; k<=3; k++)
				c[i][j]=(c[i][j]+a[i][k]*a[k][j])%p;
	for(long long i=1; i<=3; i++)
		for(long long j=1; j<=3; j++)
			a[i][j]=c[i][j];
}
void ksm(int k)
{
	ans[1][1]=ans[2][2]=ans[3][3]=1;
	while(k!=0)
	{
		if(k&1)
			multi();
		multi1();
		k>>=1;
	}
}
void multi2()
{
	long long c[10][10];
	for(long long i=1; i<=1; i++)
		for(long long j=1; j<=3; j++)
			c[i][j]=0;
	for(long long i=1; i<=1; i++)
		for(long long j=1; j<=3; j++)
			for(long long k=1; k<=3; k++)
				c[i][j]=(c[i][j]+a[i][k]*ans[k][j])%p;
	for(long long i=1; i<=1; i++)
		for(long long j=1; j<=3; j++)
			ans[i][j]=c[i][j];
}
int main()
{
	scanf("%d", &n);
	a[1][1]=a[1][2]=a[2][1]=a[3][1]=a[3][3]=1;
	ksm(n-1);
	a[1][1]=a[1][2]=a[1][3]=1;
	multi2();
	printf("%d", ans[1][2]);
	return 0;
}