POJ-3070-Fibonacci

POJ-3070-Fibonacci

传送门

这道题是经典的矩阵快速幂~
最近在写数论啦~

矩阵快速幂的经典运用。是斐波那契数列上面。
直接递推是无法完成的~
所以用矩阵表示Fibonacci数列的递推关系，然后把问题转换为求这个矩阵的多少次幂。

这道题目已经把递推关系转化为矩阵了所以。。简直了。。
小白选手（比如我）都会hhh。
矩阵快速幂的运算部分其实和数的快速幂简直不要太像好嘛。。
只不过是把数改成了矩阵的运算形式。
然后npow（）部分的初始化部分对于数是res = 1;那么在矩阵部分就是初始化res矩阵为单位矩阵即可~

把递推关系式转化为矩阵是这样的：
f(0) = 0, f(1) = 1, f(2) = 1, f(3) = 2, f(4) = 3, f(5) = 5, f(6) = 8, f(7) = 13…

所以实现起来就很简单~

代码部分就是矩阵快速幂的经典模板：

#include <iostream>
#include <cstdio>
#include <cstring>
#define mst(a, n) memset(a, n, sizeof(a))
using namespace std;
typedef long long ll;
const int mod = 1e4;

int n;
struct Matrix
{
	int m[2][2];
	Matrix()
	{
		mst(m, 0);
	}
};

Matrix Multi(Matrix a, Matrix b)
{
	Matrix res;
	for (int i = 0; i < 2; i++)
	{
		for (int j = 0; j < 2; j++)
		{
			for (int k = 0; k < 2; k++)
			{
				res.m[i][j] = (res.m[i][j] + a.m[i][k] * b.m[k][j]) % mod;
			}
		}
	}
	return res;
}

int npow(Matrix a, int n)
{
	Matrix res;
	for (int i = 0; i < 2; i++)
	{
		res.m[i][i] = 1;
	}
	while (n)
	{
		if (n & 1)
		{
			res = Multi(res, a);
		}
		a = Multi(a, a);
		n >>= 1;
	}
	return res.m[0][1];
}

int main()
{
	while (~scanf ("%d", &n) && n != -1)
	{
		Matrix a;
		a.m[0][0] = 1;
		a.m[0][1] = 1;
		a.m[1][0] = 1;
		int ans = npow(a, n);
		cout << ans << endl;
	}
	return 0;
}