M斐波那契数列

HDU - 4549

矩阵快速幂加费马小定理

---

• 链接：HDU-4549

思路：

是不以为位这样就OK了?
很遗憾，它超时了！ Time Limit Exceeded！

---

看一下这道题的数据量:
每组数据占一行，包含3个整数a, b, n（ 0 <= a, b, n <= 10^9 ）
mod=1000000007=1e9+7
我们来简单计算一下时间复杂度

• 2^10=1024 所以单纯矩阵快速幂用的时间为log(10^9)*8(三个for循环)=240
• 整数快速幂消耗的时间 log(f(10^9)) 没错就是在这TLE了，接下来想办法优化它

---

费马定理优化 次方 及证明：

咱知道最后的结果是qui(a,f(n-1))*qui(b,f(n))%mod;
也就是a^f(n-1) * b^fn%mod;
这里的f(n-1),fn太大了，得想办法让它变小
这时候费马定理闪亮登场！
费马定理：b^(p-1)%p同余1%p(b,p互质)

---

所以a,b<=1^9 p=mod=1e9+7
这里a,b都小于p,然后p是一个质数，所以a,b不可能是p的倍数，p是质数也不可能是a,b的倍数，所以a,b底数和p互质
所以可以对fn取模(p-1);
这样fn<p-1
整数快速幂的时间复杂度也就成为了Log(1e9)=30;
这样就解决问题了！！！

---

代码：

#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
struct sta{
	ll ma[3][3];
}I,Z;
const int mod=1000000007;
struct sta operator*(struct sta&a,struct sta&b)
{
	struct sta res=Z;
	for(int i=1;i<=2;i++)
	{
		for(int k=1;k<=2;k++)
		{
			for(int j=1;j<=2;j++)
			{
	res.ma[i][j]+=(a.ma[i][k]*b.ma[k][j])%(mod-1);
	
			}
		}
	}
	return res;
}

ll qui(ll a,ll n)
{
	ll res=1;
	n%=(mod-1);
	while(n)
	{
		if(n&1) res=(res*a)%mod;
		a=(a*a)%mod;
		n>>=1;
	}
	return res;
}
struct sta staqui(struct sta a,int n)
{
	struct sta res=I;
	while(n)
	{
		if(n&1) res=res*a;
		a=a*a;
		n>>=1;
	}
	return res;
}
ll a,b,n;
int main()
{
	for(int i=1;i<=2;i++)
	{
		for(int j=1;j<=2;j++)
		{
			if(i==j) I.ma[i][j]=1;
			else
			{
				I.ma[i][j]=Z.ma[i][j]=0;
			}
		}
	}
	struct sta p;
	p.ma[1][1]=1;
	p.ma[1][2]=1;
	p.ma[2][1]=1;
	p.ma[2][2]=0;
	
	//fn+1 fn-> f2 f1 -> 1 1
	//Fn=a^fn*b^f(n-1)
	
	while(cin>>a>>b>>n)
	{
		if(n==0||n==1||n==2)
		{
		if(n==0) {
			cout<<a%mod<<endl;
			continue;
		}
		if(n==1)
		{
			cout<<b%mod<<endl;
			continue;
		}
		if(n==2)
		{
			ll res=(a%mod*b%mod)%mod;
			cout<<res<<endl;
			continue;
		}
		}
	struct sta fn=staqui(p,n-3);
	struct sta fm=staqui(p,n-2);
	ll res=qui(a,(fn.ma[1][1]+fn.ma[1][2])%(mod-1))%mod*qui(b,(fm.ma[1][1]+fm.ma[1][2])%(mod-1))%mod;
	cout<<res%mod<<endl;
	}
	return 0;
}