矩阵快速幂 算法了解一下

矩阵快速幂算法,适用于指数特别大的数据。
时间复杂度可以降到O(logn)
方法:
1 构造出初始矩阵
2 用关键代码

while(n)	
	{
		if(n&1) ans=matmul(ans,a);
		a=matmul(a,a);
		n=n/2;//也可以用n=n>>1; 
	}

下面是例题
:数列第N项 20’
描述
a_n = (3a_{n-1} + 4a_{n-2})\ mod\ 1e9 + 7
a
n

=(3a
n−1

+4a
n−2

) mod 1e9+7

输入
输入三个正整数n,a_1,a_2n,a
1

,a
2

输出
输出a_na
n

样例
输入复制
5 1 1
输出复制
103
提示
数据规模

对于50%50%的数据,3\le n\le 50, 1\le a_1,a_2 \le 103≤n≤50,1≤a
1

,a
2

≤10

对于70%70%的数据,3\le n\le 10^7, 1\le a_1,a_2 \le 103≤n≤10
7
,1≤a
1

,a
2

≤10

对于100%100%的数据,3\le n\le 10^{12}, 1\le a_1,a_2 \le 103≤n≤10
12
,1≤a
1

,a
2

≤10

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
#define ll long long
const ll maxx=1e9+7;
ll n;
ll ans;


class mat
{
	public:
		ll a[2][2];
	mat()
	{
		memset(a,0,sizeof(a));
	}
};

mat & matmul(mat& aa,mat& bb)
{
	mat temp=mat();
	for(int i=0;i<2;i++)
		for(int j=0;j<2;j++)
		{
			for(int k=0;k<2;k++)
			{
				temp.a[i][j]=(temp.a[i][j]+aa.a[i][k]*bb.a[k][j])%maxx;
			}
		}
	aa=temp;
	return aa;
}


mat pow_mat(mat a,ll n)
{
	mat ans=mat();
	ans.a[0][0]=1;
	ans.a[1][1]=1;
	while(n)	
	{
		if(n&1) ans=matmul(ans,a);
		a=matmul(a,a);
		n=n/2;//也可以用n=n>>1; 
	}
	return ans;
}

int main()
{
	mat aa=mat();
	aa.a[0][0]=3;
	aa.a[0][1]=4;
	aa.a[1][0]=1;
	ll a1,a2;
	cin>>n>>a1>>a2;
	if(n==1) cout<<a1;
	else if(n==2) cout<<a2;
	else
	{
		mat daan=pow_mat(aa,n-2);
		
		ans=(a2*daan.a[0][0]+a1*daan.a[0][1])%maxx;
		cout<<ans;
	}
	
	
 	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值