hdu 4549(矩阵优化递推,数论,二分快速幂)

设f(0)=1,f(1)=0,f(n)=f(n-1)+f(n-2)(n>=2)。

那么有F(n)=a^f(n)*b^f(n+1)。

根据费马小定理,1000000007是一个素数,那么所求可以化简为a^(f(n)%1000000006)*b^(f(n+1)%1000000006),

那么就可以先求f(n)和f(n+1),再用二分快速幂求得结果。

由于n比较大,求f(n)和f(n+1)要用矩阵来优化。

讲递推式转化为

a(n)=a(n-1)+b(n-1);

b(n)=a(n-1);

得到B矩阵为{1,1,

                     1,0 }

初始矩阵为{f(1),f(0)}={1,0};

具体实现如下

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
#define ll long long
#define mod 1000000007
#define N 2
ll a,b,n;
void mul(ll a[N][N],ll b[N][N])//矩阵乘法
{
	ll ret[N][N]={0};
	int i,j,k;
	for(i=0;i<N;i++)
	{
		for(j=0;j<N;j++)
		{
			for(k=0;k<N;k++)
			{
				ret[i][j]+=a[i][k]*b[k][j];
				ret[i][j]%=mod-1;
			}
		}
	}
	for(i=0;i<N;i++)
	{
		for(j=0;j<N;j++)a[i][j]=ret[i][j];
	}
}
ll pm(ll x,ll y)//二分快速幂
{
	ll ret=1,temp=x;
	while(y)
	{
		if(y&1)ret=ret*temp%mod;
		temp=temp*temp%mod;
		y>>=1;
	}
	return ret;
}
int main()
{
	while(scanf("%I64d%I64d%I64d",&a,&b,&n)!=EOF)
	{
		ll ans[N][N]= {0,1,
			           0,0,};
		ll temp[N][N]={1,1,
			           1,0,};
		while(n)//矩阵快速幂
		{
			if(n&1)
				mul(ans,temp);
			mul(temp,temp);
			n>>=1;
		}
		printf("%I64d\n",pm(a,ans[0][1])*pm(b,ans[0][0])%mod);
	}
	return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值