【专题】矩阵快速幂

HDU-1757

思路:

这题是矩阵快速幂模板题。

代码:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int k, mod;
int num[10];
struct matrix{
	int m[10][10];
	void init_0(){
		memset(m, 0, sizeof(m));
	}
	void init_1(){
		init_0();
		for(int i = 0; i < 10; ++i)
			m[i][i] = 1;
	} 
	void init(){
		init_0();
		m[0][0] = num[0], m[0][1] = num[1], m[0][2] = num[2], m[0][3] = num[3], m[0][4] = num[4], m[0][5] = num[5], m[0][6] = num[6], m[0][7] = num[7], m[0][8] = num[8], m[0][9] = num[9];
		m[1][0] = 1;
		m[2][1] = 1;
		m[3][2] = 1;
		m[4][3] = 1;
		m[5][4] = 1;
		m[6][5] = 1;
		m[7][6] = 1;
		m[8][7] = 1;
		m[9][8] = 1;
	}
	
	matrix operator*(const matrix &M)const{
		matrix temp;
		temp.init_0();
		for(int i = 0; i < 10; ++i){
			for(int j = 0; j < 10; ++j){
				for(int k = 0; k < 10; k++){
					temp.m[i][j] = (temp.m[i][j] + (m[i][k] * M.m[k][j])%mod) % mod;
				}
			}
		}
		return temp;
	}
};

matrix qpow(int t){
	matrix ans, p;
	ans.init_1();
//	cout << "ans init complete" << endl;
	p.init();
//	cout << "enter qpow" << endl;
	while(t){
		if(t&1)ans = ans * p;
		p = p * p;
		t >>= 1;
	}
	return ans;
}

int main(){
	int init[10] = {9,8,7,6,5,4,3,2,1,0};
	while(cin >> k >> mod){
		for(int i = 0; i < 10; ++i){
			cin >> num[i];
		} 
//		cout << "input complete" << endl;
		if(k>=10){
//			cout << "k>=10" << endl;
			matrix ans = qpow(k-9);
			ll a = 0;
			for(int i = 0; i < 10; ++i){
				a = a%mod + (ans.m[0][i]%mod) * (init[i]%mod);
			}
			cout << a%mod << endl;
		}
		else{
			cout << k%mod << endl;
		}
	}
	return 0;
} 

 

HDU-1575

思路:

矩阵快速幂模板题。

求完快速幂后把主对角线的加起来就是答案。

代码:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod = 9973;
int t, n, k;
int a[11][11];
struct matrix{
	int m[11][11];
	void init_0(){
		memset(m, 0, sizeof(m));
	}
	void init_1(){
		init_0();
		for(int i = 1; i <= n; ++i){
			m[i][i] = 1;
		}
	}
	void init(){
		for(int i = 1; i <= n; ++i){
			for(int j = 1; j <= n; ++j){
				m[i][j] = a[i][j];
			}
		}
	}
	matrix operator*(matrix &M)const{
		matrix temp;
		temp.init_0();
		for(int i = 1; i <= n ; ++i){
			for(int j = 1; j <= n; ++j){
				for(int k = 1; k <= n; ++k){
					temp.m[i][j] = (temp.m[i][j] + (m[i][k] * M.m[k][j])%mod)%mod;
				}
			}
		}
		return temp;
	} 
};

matrix qpow(ll k){
	matrix ans, p;
	ans.init_1();
	p.init();
	while(k){
		if(k&1) ans = ans * p;
		p = p * p;
		k >>= 1;
	}
	return ans;
}

int main(){
	cin >> t;
	while(t--){
		cin >> n >> k;
		for(int i = 1; i <= n; ++i){
			for(int j = 1; j <= n; ++j){
				cin >> a[i][j];
			}
		}
		matrix ans = qpow(k);
		ll sum = 0;
		for(int i = 1; i <= n; ++i){
			sum += ans.m[i][i];
		}
		cout << sum%mod << endl;
	}
	return 0;
}

 

HDU-2604

思路:

算出前面几个值后可以看出 n\geq 4 时,F(N)=F(n-1)+F(n-3)+F(N-4)

根据公式就可以写出转移矩阵了。

代码:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int n, mod;
struct matrix{
	ll m[11][11];
	void init_0(){
		memset(m, 0, sizeof(m));
	}
	void init_1(){
		init_0();
		for(int i = 1; i <= 4; ++i){
			m[i][i] = 1;
		}
	}
	void init(){
		init_0();
		m[1][1] = 1, m[1][3] = 1, m[1][4] = 1;
		m[2][1] = 1;
		m[3][2] = 1;
		m[4][3] = 1; 
	}
	matrix operator*(matrix &M)const{
		matrix temp;
		temp.init_0();
		for(int i = 1; i <= 4; ++i){
			for(int j = 1; j <= 4; ++j){
				for(int k = 1; k <= 4; ++k){
					temp.m[i][j] = (temp.m[i][j] + (m[i][k] * M.m[k][j])%mod)%mod;
				}
			}
		}
		return temp;
	} 
};

matrix qpow(ll k){
	matrix ans, p;
	ans.init_1();
	p.init();
	while(k){
		if(k&1) ans = ans * p;
		p = p * p;
		k >>= 1;
	}
	return ans;
}

int main(){
	int a[5] = {0, 1, 2, 4, 6};
	while(cin >> n >> mod){
		if(n<=3){
			cout << a[n+1]%mod << endl;;
		}
		else {
			ll sum = 0;
			matrix ans = qpow(n-3);
			for(int i = 1; i <= 4; ++i){
				sum += ans.m[1][i] * a[4-i+1] % mod;
			}
			cout << sum % mod << endl;
		}
	}
	return 0;
}

 

HDU-2256

思路:

代码:

#include <cstdio>
#include <cstring>
using namespace std;
typedef long long ll;
const int mod = 1024;
int t;
ll n;

struct matrix{
	int m[3][3];
	
	void init_0(){
		memset(m, 0, sizeof(m));
	}
	
	void init_1(){
		init_0();
		for(int i = 1; i <= 2; ++i){
			m[i][i] = 1;
		}
	}
	
	void init(){
		init_0();
		m[1][1] = 5, m[1][2] = 12;
		m[2][1] = 2, m[2][2] = 5;
	}
	
	matrix operator*(const matrix &M)const{
		matrix ans;
		ans.init_0();
		for(int i = 1; i <= 2; ++i){
			for(int j = 1; j <= 2; ++j){
				for(int k = 1; k <= 2; ++k){
					ans.m[i][j] = (ans.m[i][j] + (m[i][k] * M.m[k][j]) % mod) % mod;
				}
			}
		}
		return ans;
	}
};
matrix a;
ll qpow(ll T){
	matrix ans, p;
	ans.init_1();
	p.init();
	while(T){
		if(T&1)ans = ans * p;
		p = p * p;
		T >>= 1;
	}
	ans = ans * a;
	return 2*ans.m[1][1] - 1;
}

int main(){
	scanf("%d", &t);
	a.m[1][1] = 5, a.m[2][1] = 2;
	while(t--){
		scanf("%lld", &n);
		if(n==1)printf("9\n");
		else printf("%lld\n", qpow(n-1)%1024);
	}
	return 0;
} 

 

CodeForces - 185A

思路:

根据观察,一个上三角形生成 3 个小的上三角形;一个下三角形生成 1 个上三角形。

推出公式 F(N)=2*F(N-1)+4^{N-1}(N\geq 1)

代码:

#include <cstdio>
#include <cstring>
using namespace std;
typedef long long ll;
const int mod = 1000000007;
ll n;

struct matrix{
	ll m[3][3];
	
	void init_0(){
		memset(m, 0, sizeof(m));
	}
	
	void init_1(){
		init_0();
		for(int i = 1; i <= 2; ++i){
			m[i][i] = 1;
		}
	}
	
	void init(){
		init_0();
		m[1][1] = 2, m[1][2] = 4;
		m[2][1] = 0, m[2][2] = 4;
	}
	
	matrix operator*(const matrix &M)const{
		matrix ans;
		ans.init_0();
		for(int i = 1; i <= 2; ++i){
			for(int j = 1; j <= 2; ++j){
				for(int k = 1; k <= 2; ++k){
					ans.m[i][j] = (ans.m[i][j] + (m[i][k] * M.m[k][j]) % mod) % mod;
				}
			}
		}
		return ans;
	}
};
matrix a;
ll qpow(ll T){
	matrix ans, p;
	ans.init_1();
	p.init();
	while(T){
		if(T&1)ans = ans * p;
		p = p * p;
		T >>= 1;
	}
	ans = ans * a;
	return ans.m[1][1];
}

int main(){
	a.m[1][1] = 3, a.m[2][1] = 1;
	scanf("%I64d", &n);
	if(n==0)printf("1\n");
	else if(n==1)printf("3\n");
	else printf("%I64d\n", qpow(n-1)%mod);
	return 0;
} 

 

HDU - 2842

思路:

首先,明确游戏规则:前 k 个环取走,第 k+1 还在时,可以取走第 k+2 个环。

例如,有 1、2、3、4、5、6 共 6 个环。

当前剩下 4、5、6,这时可以取走 5 号环。

即,要取走前 k 个环(F(K)),要取走前 K-2 个环(F(K-2))后,拿走第 K 个环,再把前 K-2 个环(F(K-2))放回去,再执行拿走前 K-1 个环(F(K-1)),最后再补上拿走第 1 个环的 1 次。

以上就可以得出递推式:F(K)=F(k-1)+2*F(k-2)+1 (K>2)

之后就可以得出转移矩阵

\begin{bmatrix} 1& 2 &1 \\ 1& 0 &0 \\ 0& 0 &1 \end{bmatrix}

初始矩阵

\begin{bmatrix} 2& 0 &0 \\ 1& 0 &0 \\ 1& 0 &0 \end{bmatrix}

代码:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod = 200907;
ll n;

struct matrix{
	ll m[5][5];
	
	void init_0(){
		memset(m, 0, sizeof(m));
	}
	void init_1(){
		init_0();
		for(int i = 1; i <= 3; ++i)m[i][i] = 1;
	}
	void init(){
		init_0();
		m[1][1] = 1, m[1][2] = 2, m[1][3] = 1;
		m[2][1] = 1;
		m[3][3] = 1;
	}
	matrix operator*(const matrix &M)const{
		matrix ans;
		ans.init_0();
		for(int i = 1; i <= 3; ++i)
			for(int j = 1; j <= 3; ++j)
				for(int k = 1; k <= 3; ++k)
					ans.m[i][j] = (ans.m[i][j] + (m[i][k] * M.m[k][j]) % mod) % mod;
		return ans;
	}
}; 
matrix a;
ll qpow(ll t){
	matrix ans, p;
	ans.init_1();
	p.init();
	while(t){
		if(t&1) ans = ans * p;
		p = p * p;
		t >>= 1;
	}
	ans = ans * a;
	return ans.m[1][1]%mod;
}

int main(){
	a.m[1][1] = 2, a.m[2][1] = 1, a.m[3][1] = 1;
	while(scanf("%lld", &n)!=EOF, n){
		if(n>=3){
			printf("%lld\n", qpow(n-2));
		}
		else{
			printf("%lld\n", n);
		}
	}
	return 0;
} 

未完待续。。。

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值