技術標籤：矩陣乘法快速冪

Time Limit:1000MS Memory Limit:65536K
Total Submit:53 Accepted:41

Description

求數列 f [ n ] = f [ n − 1 ] + f [ n − 2 ] + n + 1 , f [ 1 ] = f [ 2 ] = 1 f[n]=f[n-1]+f[n-2]+n+1,f[1]=f[2]=1 f[n]=f[n−1]+f[n−2]+n+1,f[1]=f[2]=1的前n項的和 s [ n ] s[n] s[n]

Input

N ( １ ＜ Ｎ ＜ 2 3 1 − 1 ) N(１＜Ｎ＜2^31-1)

Output

第n項結果

Sample Input

100

Sample Output

2528

Source

elba

解題思路

矩陣乘法

考慮1×5的矩陣 【 f [ n − 2 ] , f [ n − 1 ] , s [ n − 2 ] , n , 1 】 【f[n-2],f[n-1],s[n-2],n,1】 【f[n−2],f[n−1],s[n−2],n,1】,
我們需要找到一個5×5的矩陣A，使得它乘以A得到如下1×5的矩陣：
【 f [ n − 1 ] , f [ n ] , s [ n − 1 ] , n + 1 , 1 】 = 【 f [ n − 1 ] , f [ n − 1 ] + f [ n − 2 ] + n + 1 , s [ n − 2 ] + f [ n − 1 ] , n + 1 , 1 】 【f[n-1],f[n],s[n-1],n+1,1】 =【f[n-1], f[n-1]+f[n-2]+n+1,s[n-2]+f[n-1],n+1,1】

程式碼

#include<algorithm>
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
const int INF=9973;
long long n;
using namespace std;
struct c{
	int n,m;
	int a[10][10];
}A,
 
B,CC;
c operator *(c A,c B){
	c C;
	C.n=A.n,C.m=B.m;
	for(int i=1;i<=C.n;i++)
		for(int j=1;j<=C.m;j++)
			C.a[i][j]=0;
	for(int k=1;k<=A.m;k++)
	{
		for(int i=1;i<=C.n;i++)
			for(int j=1;j<=C.m;j++)
				C.a[i][j]=(C.a[i][j]+(A.a[i][k]*B.a[k][j])%INF)%INF;
	}	
	return C;
}
void poww(long long x){
	if(x==1)
	{
		B=A;
		return; 
	}
	poww(x>>1);
	B=B*B;
	if(x&1)
		B=B*A;
	
}
int main(){
	scanf("%lld",&n);
	if(n==1){
		printf("1");
		return 0;
	}
	A.n=5,A.m=5;
	A.a[1][1]=0,A.a[1][2]=1,A.a[1][3]=0,A.a[1][4]=0,A.a[1][5]=0;
	A.a[2][1]=1,A.a[2][2]=1,A.a[2][3]=0,A.a[2][4]=0,A.a[2][5]=1;
	A.a[3][1]=0,A.a[3][2]=1,A.a[3][3]=1,A.a[3][4]=0,A.a[3][5]=0;
	A.a[4][1]=0,A.a[4][2]=1,A.a[4][3]=1,A.a[4][4]=1,A.a[4][5]=0;
	A.a[5][1]=0,A.a[5][2]=0,A.a[5][3]=0,A.a[5][4]=0,A.a[5][5]=1;
	poww(n-1);
	CC.n=1,CC.m=5;
	CC.a[1][1]=1,CC.a[1][2]=1,CC.a[1][3]=3,CC.a[1][4]=1,CC.a[1][5]=1;
	CC=CC*B;
	printf("%d\n",CC.a[1][5]);
	
}