幼儿园数学题II

幼儿园数学题II

题目链接：SSL 2514

题目大意

有公式 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。
要你求前 n 项和。

思路

这道题我们可以化公式变成 f(n)=f(n-1)+f(n-2)+n+1。

首先，我们可以移项得到：f(n)=f(3)+f(4)+f(5)+…+f(n-3)+f(n-2)+(n+4)(n-1)/2
然后把后面的那个拆开，得到：f(n)=f(3)+f(4)+f(5)+…+f(n-3)+f(n-2)+n^2/2+3n/2-2

这是 f(n)，然后我们看 f(n+1)，可以通过同样的方式得到是：f(n+1)=f(3)+f(4)+f(5)+…+f(n-3)+f(n-2)+f(n-1)+n^2/2+5n/2

可以发现，相比于 f(n)，多了 f(n-1) + n + 2。

那就是 f(n+1)=f(n)+f(n-1)+n+2。
然后可以变成 f(n+1)=f(n)+f(n-1)+(n+1)+1。
然后如果要求 f(n)，那就是 f(n)=f(n-1)+f(n-2)+n+1。（就是把 n+1 全部变成 n）

然后就得到了我们要的。

然后就用矩阵乘法算，之前写过一样的公式。
——>点我查看<——

代码

#include<cstdio>
#define mo 1000000007
#define ll unsigned long long

using namespace std;

struct matrix {
	ll n, m, a[5][5];
}a, b, ans, re;
ll n;

matrix operator *(matrix x, matrix y) {
	re.n = x.n;
	re.m = y.m;
	for (ll i = 0; i < re.n; i++)
		for (ll j = 0; j < re.m; j++)
			re.a[i][j] = 0;
	
	for (ll k = 0; k < x.m; k++)
		for (ll i = 0; i < re.n; i++)
			for (ll j = 0; j < re.m; j++)
				re.a[i][j] = (re.a[i][j] + (x.a[i][k] * y.a[k][j]) % mo) % mo;
	
	return re;
}

void jzksm(ll now) {
	b = a;
	now--;
	while (now) {
		if (now & 1) b = b * a;
		a = a * a;
		now /= 2;
	}
}

int main() {
	scanf("%lld", &n);
	
	if (n == 1) {
		printf("1");
		return 0;
	}
	
	ans.n = 1;
	ans.m = 5;
	ans.a[0][0] = 1;
	ans.a[0][1] = 1;
	ans.a[0][2] = 3;
	ans.a[0][3] = 1;
	ans.a[0][4] = 1;
	
	a.n = 5;
	a.m = 5;
	a.a[1][0] = 1;
	a.a[0][1] = 1;a.a[1][1] = 1;a.a[2][1] = 1;a.a[3][1] = 1;
	a.a[2][2] = 1;a.a[3][2] = 1;
	a.a[3][3] = 1;
	a.a[1][4] = 1;a.a[4][4] = 1;
	
	jzksm(n - 1);
	
	ans = ans * b;
	
	printf("%lld", ans.a[0][4]);
	
	return 0;
}