【SSL2514】幼儿园数学题II

Description

这天,当一头雾水的LZH同学在考场上痛哭的时候,一旁的YMW早就如切菜一样cut掉了简单至极的第一题,风轻云淡的冲击着满分,然而最后一道题着实难道了他,毕竟是幼儿园副园长树皮和著名毒瘤秋彪为了防止人AK而出的,可是YMW作为ACrush的著名粉丝,向来以AK为目标,永不言败,而他能不能AK就看你了
题目是酱紫的,f(n)-f(3)-f(4)-f(5)-…-f(n-3)-f(n-2)=(n+4)(n-1)/2,f(1)=1,f(2)=1
求f(n)的前n项和

Input

输入 一个正整数n（保证0<=n<=2^31-1）

Output

输出 一个正整数,表示这个图形的整点个数,需要对1000000007求余

Sample Input

样例输入1
1
样例输出1
1
样例输入2
2
样例输出2
2

思路：

我们考虑分解式子：
我们直接从5开始分解
f(5)=f(5)
f(6)=f(3)+f(4)+25
f(7)=f(3)+f(4)+f(5)+33=f(6)+f(5)+7+1
f(8)=f(3)+f(4)+f(5)+f(6)+42=f(7)+f(6)+8+1
……
由此可得，题目就是让你求s(n)，其中f(n)=f(n-1)+f(n-2)+n+1
直接矩阵乘法
转移矩阵请参考矩阵乘法例5

code:

#include<iostream>
#include<cstdio>
using namespace std;
long long n;
long long a[10][10];
long long ans[10][10];
long long p=1000000007;
void multi()
{
	long long c[10][10];
	for(long long i=1; i<=5; i++)	
		for(long long j=1; j<=5; j++)
			c[i][j]=0;
	for(long long i=1; i<=5; i++)
		for(long long j=1; j<=5; j++)	
			for(long long k=1; k<=5; k++)	
				c[i][j]=(c[i][j]+ans[i][k]*a[k][j])%p;
	for(long long i=1; i<=5; i++)	
		for(long long j=1; j<=5; j++)
			ans[i][j]=c[i][j];
}
void multi1()
{
	long long c[10][10];
	for(long long i=1; i<=5; i++)	
		for(long long j=1; j<=5; j++)
			c[i][j]=0;
	for(long long i=1; i<=5; i++)
		for(long long j=1; j<=5; j++)	
			for(long long k=1; k<=5; k++)	
				c[i][j]=(c[i][j]+a[i][k]*a[k][j])%p;
	for(long long i=1; i<=5; i++)	
		for(long long j=1; j<=5; j++)
			a[i][j]=c[i][j];
}
void ksm(long long k)
{
	ans[1][1]=ans[2][2]=ans[3][3]=ans[4][4]=ans[5][5]=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<=5; j++)
			c[i][j]=0;
	for(long long i=1; i<=1; i++)
		for(long long j=1; j<=5; j++)	
			for(long long k=1; k<=5; 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<=5; j++)
			ans[i][j]=c[i][j];
}
int main()
{
	scanf("%lld", &n);
	if(n==1)
	{
		printf("1");
		return 0;
	}
	a[1][1]=a[2][1]=a[4][1]=a[5][1]=1;
	a[1][2]=1;
	a[1][3]=a[3][3]=1;
	a[4][4]=a[5][4]=1;
	a[5][5]=1;
	ksm(n-2);
	a[1][1]=a[1][2]=1;
	a[1][3]=1;
	a[1][4]=3;
	a[1][5]=1;
	multi2();
	printf("%lld", (ans[1][1]+ans[1][3])%p);
	return 0;
}