矩阵快速幂(个人心得+白书板子)

矩阵快速幂的大体思想与快速幂想法一致,只是普通的数乘,变成了矩阵相乘,所以问题的关键在于矩阵乘法上,即(看代码)

ll pow(ll x,ll n,ll mod){
	ll res=1;//此处的1变为单位阵 
	while(n>0){
		if(n&1)res=res*x%mod;//res*x变成矩阵乘法 
		x=x*x%mod;//同上 
		n>>=1;
	}
	return res;
} 

 矩阵乘法板子(个人认为这是矩阵快速幂的关键):

typedef vector<int> vec;
typedef vector<vec> mat;

mat mul(mat &A,mat&B){
	mat C(A.size(),vec(B[0].size()));
	for(int i=0;i<A.size();i++){
		for(int k=0;k<B.size();k++){
			for(int j=0;j<B[0].size();j++){
				C[i][j]=(C[i][j]+A[i][k]*B[k][j])%MOD;
			}
		}
	}
	return C;
}

 矩阵快速幂(核心):

mat pow(mat A,ll n){
	mat B(A.size(),vec(A.size()));
	//单位阵 I 
	for(int i=0;i<A.size();i++){
		B[i][i]=1;
	}
	while(n>0){
		if(n&1)B=mul(B,A);
		A=mul(A,A);
		n>>=1;
	}
	return B;
}

 

posted @ 2018-05-31 21:52  visualVK  阅读(250)  评论(0编辑  收藏  举报