C++快速幂,矩阵快速幂(模板)

2 篇文章 0 订阅
1 篇文章 0 订阅

目标:矩阵快速幂 掌握 

需要技能:矩阵乘法,c语言基础

要求能力:和我一样弱都能学会。

矩阵乘法大概就是:左行*右列(左边的矩阵第一行×右边第一列然后求和 就是 乘法的第一行第一个元素)

 


左边是行 右边是列 横是行 竖着是列

2行2列 * 2行2列 = 2行2列

 红框看为第一行和第一列 代表 答案的 第一行第一个元素

 绿框看为第一行和第二列 代表 答案的 第一行第二个元素

 蓝框看为第二行和第一列 代表 答案的 第二行第一个元素

 紫框看为第二行和第二列 代表 答案的 第二行第二个元素

#include<bits/stdc++.h> //万能头 
using namespace std;//namespace  
typedef long long ll;//定义ll 为long long  方便使用 
#define N 111//最大数 
const int M  = 1e9+7;//Mod 需要模这个数 
#define mod(x) ((x)%M )//定义 mod的函数 
int n ;
struct mat{
	ll m[N][N];
}unit;//矩阵结构体 以及单位矩阵unit (主对角为1其余为0)
mat operator  * (mat a,mat b){ //重载operator 乘法  
	mat ret ;
	ll val,i,j;
	for(i =  0 ; i < n ; i++)
	{
		for(j= 0 ; j < n ; j++){
			val = 0;
			for(int k = 0 ; k < n;k++){
				val+= mod((ll)a.m[i][k]*b.m[k][j]);  //三个循环是求矩阵乘法 mod一下 
			}
			ret.m[i][j] = mod(val);//也需要mod 一下 都是为了防止数太大 
		}
		
	}
	return ret;//返回 俩个矩阵乘法的结果 
}
void init(){
	for(int i = 0 ; i < N ;i++){
		unit.m[i][i] = 1;
	} //init 单位矩阵 
}
mat quick_pow(mat  a, ll n){//快速矩阵幂  
	mat res = unit ;
	while(n){
		if(n&1)	res=res*a;
		a =  a*a;
		n>>=1;
	}
	return res;
}
int main(){ 
	cin >> n ;
	init();
	mat a,b;	
	cout<<"第一个"<<endl;
	for(int i = 0 ; i < n ; i++){
		for(int j = 0 ; j <n ;j++){
			cin>>a.m[i][j];
		}
	}
	//写入 矩阵  
	mat ans  =	quick_pow(a,2);  //调用快速矩阵  2是幂 
	for(int i = 0 ; i < n ; i++){
		for(int j = 0 ; j <n ;j++){
			cout<<ans.m[i][j]<<"  ";
		}
		cout<<endl;
	}//输出结果 
	return 0;
}

如果quick_pow看不懂,看看int如何快速幂,博客有介绍。

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值