幼儿园数学题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;
}