前n项的和【矩阵乘法】

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$的前n项的和$s[n]$

---

Input

$N(\mathrm{１}＜Ｎ＜2^{3}1-1)$

Output

第n项结果

---

Sample Input

100

Sample Output

2528

---

Source

elba

---

解题思路

矩阵乘法

考虑1×5的矩阵$【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】$
容易构造出A为：

---

代码

#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]);
	
}