斐波那契数列III

斐波那契数列III

题目链接：SSL 1530

题目大意

就是定义一个函数：$f_i=f_{i-1}+f_{i-2}+1$，问你某一项是什么。
注：$f_1=f_2=1$

思路

这一题我们只需要在斐波那契数列II上稍稍改动矩阵，就可以A。

这一道题每次都要多加一个 $1$，那我们矩阵就可以这样：
原来 $1\times 2$ 的变成 $1\times 3$ 的：

1(F[i-2])	1(F[i-1])	1(每次要加的1)

然后原来那个 $2\times 2$ 的转移矩阵就变成了 $3\times 3$ 的：

0	1	0
1	1	0
0	1	1

然后改一下模数，也是把 $n-1$ 个转移矩阵乘起来，然后再拿 $1\times 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;
}