【2019牛客暑期多校训练营第五场 B题】【generator 1】【大数矩阵快速幂】

最新推荐文章于 2020-05-19 22:50:56 发布

linbinwu123

最新推荐文章于 2020-05-19 22:50:56 发布

阅读量158

点赞数

分类专栏： # 矩阵快速幂其他文章标签： ACM

本文链接：https://blog.csdn.net/weixin_44282912/article/details/98103290

版权

其他同时被 2 个专栏收录

5 篇文章 0 订阅

订阅专栏

矩阵快速幂

1 篇文章 0 订阅

订阅专栏

题目链接：
https://ac.nowcoder.com/acm/contest/885/B
题意：
已知 $x_n=ax_{n-1}+bx_{n-2}$ ，给你 $x_0,x_1,a,b,n,mod$ ，让你求 $x_n\%mod$ 的值
题解：
这种递推式很容易想到用矩阵乘法来计算，
一般构造方法都是这样的：
$x_n=ax_{n-1}+bx_{n-2}$
$x_{n-1} = x_{n-1}$
就是多写上面这看似没用的一项就可以用矩阵写起来
$\begin{pmatrix} x_n\\ x_{n-1}\end{pmatrix}=\begin{pmatrix} a&b\\ 1&0\end{pmatrix}\begin{pmatrix} x_{n-1}\\ x_{n-2}\end{pmatrix}$
最后一项项递推就得到
$\begin{pmatrix} x_{n+1}\\ x_{n}\end{pmatrix}=\begin{pmatrix} a&b\\ 1&0\end{pmatrix}^n\begin{pmatrix} x_{1}\\ x_{0}\end{pmatrix}$
可以看到我们要求的 $x_n$ 就是左下角那个元素
但是这里有一个问题，这里的n非常大，有 $10^{10^6}$ 级别，就是1e6位的数。
原来的矩阵快速幂是 $log_2$ 的，这里要优化到 $l g$ 级别的，就是以10为底。
具体可以看代码。
代码：

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
#define sz sizeof
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> Pair;
const int MAX = 1e3 + 10;

ll x0, x1, a, b, mod;
string s;

struct Matrix {
	static const int N = 2;
	ll a[2][2];
	Matrix() {
		a[0][0] = a[1][1] = 1;
		a[0][1] = a[1][0] = 0;
	}
	Matrix operator * (const Matrix& A) const {//重载矩阵乘法
		Matrix t;
		memset(t.a, 0, sz(t.a));
		for (int i = 0; i < N; i++)
			for (int j = 0; j < N; j++)
				for (int k = 0; k < N; k++)
					t.a[i][j] = (t.a[i][j] + A.a[i][k] * a[k][j]) % mod;
		return t;
	}
};

Matrix power(Matrix A, int k) {//普通的快速幂
	if (!k) return Matrix();
	Matrix res;
	while (k) {
		if (k & 1)res = res * A;
		A = A * A;
		k >>= 1;
	}
	return res;
}

Matrix power(Matrix A, string s) {//优化的快速幂
	Matrix res;
	for (int i = s.length() - 1; i >= 0; --i) {
		if (s[i] != '0')res = res * power(A, s[i] - '0');
		A = power(A, 10);
	}
	return res;
}

int main() {
	ios::sync_with_stdio(0);
	cin >> x0 >> x1 >> a >> b;
	cin >> s >> mod;
	Matrix A;
	A.a[0][0] = a, A.a[0][1] = b, A.a[1][0] = 1, A.a[1][1] = 0;
	A = power(A, s);
	ll ans = ((A.a[1][0] * x1) % mod + (A.a[1][1] * x0) % mod) % mod;
	cout << ans << endl;
	return 0;
}