矩阵快速幂【模板】

最新推荐文章于 2024-08-21 21:14:30 发布

一幻千秋

最新推荐文章于 2024-08-21 21:14:30 发布

阅读量589

点赞数 11

分类专栏：模板题，牛客文章标签：矩阵算法 c++ c语言数据结构

本文链接：https://blog.csdn.net/CSDN2151530018/article/details/140951367

版权

模板题，牛客专栏收录该内容

17 篇文章 0 订阅

订阅专栏

链接：登录—专业IT笔试面试备考平台_牛客网
来源：牛客网

题目描述

已知数列f(n) f(n)\spacef(n) 满足 f(0) =x,f(1)=y,f(n)=af(n−1)+bf(n−2)f(0)\space = x, f(1)=y, f(n)=af(n-1)+bf(n-2)f(0) =x,f(1)=y,f(n)=af(n−1)+bf(n−2)。

输入x,y,a,b,n，求f(n)对109+710^9+7109+7取模的结果。

输入描述:

第一行输入一个整数T，表示T组测试用例。

接下来T行，每行5个数,x,y,a,b,n

1≤T≤1041 \le T \le 10^41≤T≤104
0≤x,y,a,b≤1090 \le x,y,a,b \le 10^90≤x,y,a,b≤109
0≤n≤10180 \le n \le 10^{18}0≤n≤1018

输出描述:

输出T行，每行一个答案。

输入：

输出：

答案：

#include<bits/stdc++.h>
using namespace std;
using ll = long long;
const int N = 1e3 + 10;
const ll mod = 1e9 + 7;
ll x, y, a, b, n;
struct M{
	ll a[3][3];
	M(){
		memset(a, 0, sizeof(a));
	}
}dis, ans;

void init(){//创建单位矩阵以及如快速幂一样的不断叠加的矩阵
	dis.a[1][1] = a; dis.a[1][2] = 1;
	dis.a[2][1] = b; dis.a[2][2] = 0;
    ans.a[1][1] = (a * y + b * x) % mod;
	ans.a[1][2] = ans.a[2][1] = y; ans.a[2][2] = x;
}
M getM(M x, M y){//矩阵的计算方法
	M temp;
	for(int i = 1; i <= 2; i++){
		for(int j = 1;j <= 2; j++){
			for(int k = 1; k <= 2; k++){
				temp.a[i][j] = (temp.a[i][j] + x.a[i][k] * y.a[k][j]) % mod;
			}
		}
	}
	return temp;
}
void power(ll p){//幂级计算判断为奇数相当于转化为二进制右移
	while(p){
		if(p % 2 == 1){
			ans = getM(ans, dis);
		}
		p /= 2;
		dis = getM(dis, dis);
		
	}
}

int main(){
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	int t; cin >> t;
	while(t--){
		cin >> x >> y >> a >> b >> n;
        init();
        power(n);
        cout << ans.a[2][2] << '\n';
	}
	return 0;
}

解释{ai}

矩阵快速幂是一种高效的算法，用于计算矩阵的高次方，其时间复杂度从朴素的O(n)降低到O(logn)。这种算法在解决一些递推关系、动态规划等问题时非常有用，特别是在需要频繁进行矩阵乘法操作时。以下是对矩阵快速幂的详细解释：

一、基本概念

矩阵快速幂是用来高效地计算矩阵的高次方的算法。在常规方法中，计算一个矩阵的n次方需要连续乘以该矩阵n-1次，但这种方法的时间复杂度较高。矩阵快速幂通过利用矩阵乘法的结合律，将计算过程优化为对数时间复杂度。

二、基本原理

矩阵快速幂的基本原理与整数快速幂类似，都是利用分治的思想。在整数快速幂中，我们将指数表示为二进制形式，然后利用指数的每一位来决定是否将当前的底数乘以自己（即平方操作），并相应地更新指数。在矩阵快速幂中，我们将这个过程应用到矩阵上，即将矩阵的幂次操作转化为矩阵的乘法操作。

具体来说，我们可以将矩阵的n次方表示为一系列矩阵乘法的结果。首先，我们将n表示为二进制形式，然后从最低位开始，依次判断每一位是否为1。如果某一位为1，则将当前的矩阵（初始时为原矩阵或单位矩阵）乘以该矩阵的相应次方（通过之前的计算得到）。然后，将当前的矩阵平方（即乘以自己），并将指数右移一位（即去掉最低位）。这个过程一直持续到指数为0为止。

三、算法步骤

初始化：将结果矩阵初始化为单位矩阵（如果n为0，则直接返回单位矩阵）。
二进制分解：将n表示为二进制形式。
循环计算：从最低位开始遍历n的二进制表示中的每一位。
- 如果当前位为1，则将结果矩阵乘以原矩阵的相应次方（通过之前的计算得到）。
- 将原矩阵平方（即乘以自己）。
- 将指数右移一位（即去掉最低位）。
返回结果：当指数为0时，循环结束，返回结果矩阵。