斐波那契数列III
题目链接:SSL 1530
题目大意
就是定义一个函数:
f
i
=
f
i
−
1
+
f
i
−
2
+
1
f_i=f_{i-1}+f_{i-2}+1
fi=fi−1+fi−2+1,问你某一项是什么。
注:
f
1
=
f
2
=
1
f_1=f_2=1
f1=f2=1
思路
这一题我们只需要在斐波那契数列II上稍稍改动矩阵,就可以A。
这一道题每次都要多加一个
1
1
1,那我们矩阵就可以这样:
原来
1
×
2
1\times 2
1×2 的变成
1
×
3
1\times 3
1×3 的:
1(F[i-2]) | 1(F[i-1]) | 1(每次要加的1) |
---|
然后原来那个 2 × 2 2\times 2 2×2 的转移矩阵就变成了 3 × 3 3\times 3 3×3 的:
0 | 1 | 0 |
---|---|---|
1 | 1 | 0 |
0 | 1 | 1 |
然后改一下模数,也是把 n − 1 n-1 n−1 个转移矩阵乘起来,然后再拿 1 × 3 1\times 3 1×3 的矩阵乘它,然后输出左边那个就可以了。
代码
#include<cstdio>
#define mo 9973//记得模数不一样
using namespace std;
struct matrix {
int n, m, a[5][5];
}a, b, ans, z;
int n;
matrix operator *(matrix x, matrix y) {
z.n = x.n;
z.m = y.m;
for (int i = 0; i < z.n; i++)
for (int j = 0; j < z.m; j++)
z.a[i][j] = 0;
for (int k = 0; k < x.m; k++)
for (int i = 0; i < z.n; i++)
for (int j = 0; j < z.m; j++)
z.a[i][j] = (z.a[i][j] + (x.a[i][k] * y.a[k][j]) % mo) % mo;
return z;
}
void jzksm(int now) {
b = a;
now--;
while (now) {
if (now & 1) b = b * a;
a = a * a;
now /= 2;
}
}
int main() {
scanf("%d", &n);
if (n == 1) {
printf("1");
return 0;
}
ans.n = 1;
ans.m = 3;
ans.a[0][0] = 1;
ans.a[0][1] = 1;
ans.a[0][2] = 1;//1*3的矩阵
a.n = 3;
a.m = 3;
a.a[0][0] = 0;
a.a[0][1] = 1;
a.a[0][2] = 0;
a.a[1][0] = 1;
a.a[1][1] = 1;
a.a[1][2] = 0;
a.a[2][0] = 0;
a.a[2][1] = 1;
a.a[2][2] = 1;//3*3的转移矩阵
jzksm(n - 1);
ans = ans * b;
printf("%d", ans.a[0][0]);
return 0;
}