【十进制矩阵快速幂】 2019牛客暑期多校训练营（第五场） B

最新推荐文章于 2024-04-28 13:01:55 发布

原创最新推荐文章于 2024-04-28 13:01:55 发布 · 212 阅读

0 ·

CC 4.0 BY-SA版权

矩阵快速幂专栏收录该内容

1 篇文章

订阅专栏

原文：https://blog.youkuaiyun.com/henucm/article/details/98492753

题意：给你x0、x1 a、b、mod，根据求出 $\large x_{n}$

根据上述公式，再套用十进制矩阵快速幂即可（二进制好像会超时...）

用十进制

设 $\large \tbinom{a \ \ \ b}{1\ \ \ 0}$ 为res 则可得 $\large ans=res^{n[i]-'0'}$

我们再设 $\large res=res^{10*(i-1)}$

就是分解n为每一位，再去相乘。

例如 $\large res^{298}=res^8*res^{10*9}*res^{100*2}$

讲计算出来的 $\large res.m[1][0]*x1+res.m[1][1]*x0$ 即可

#include <bits/stdc++.h>
using namespace std;
#define endl '\n'
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
//typedef __int128 bll;
const ll maxn = 1e6+100;
//const ll mod = 1e9+7;
const ld pi = acos(-1.0);
const ll inf = 1e18;
const ld eps = 1e-5;
const ld e = exp(1);

ll x0,x1,a,b,mod;
char str[maxn];

struct mat
{
	ll m[2][2];
};

mat mul(mat a,mat b)
{
	mat res;
	for(int i = 0; i < 2; i++)
		for(int j = 0; j < 2; j++)
			res.m[i][j] = 0;	
	
	for(int i = 0; i < 2; i++)
	{
		for(int j = 0; j < 2; j++)
		{
			for(int k = 0; k < 2; k++)
			{
				res.m[i][j] = (res.m[i][j]%mod + a.m[i][k]*b.m[k][j]%mod )%mod;
			}
		}
	}
	
	return res;
	
}

mat qpow(mat a,int b)
{
	mat e;
	
	for(int i = 0; i < 2; i++)
	{
		for(int j = 0; j < 2; j++)
		{
			i == j? e.m[i][j] = 1 : e.m[i][j] = 0;
		}
	}
	
	while(b)
	{
		if(b&1)
			e = mul(e,a);		
		a = mul(a,a);
		b = b >> 1;
	}
	return e;
}

mat qpow10()
{
	mat x,e;
	x.m[0][0] = a;	x.m[0][1] = b;
	x.m[1][0] = 1;  x.m[1][1] = 0;
	
	for(int i = 0; i < 2; i++)
	{
		for(int j = 0; j < 2; j++)
		{
			i == j? e.m[i][j] = 1 : e.m[i][j] = 0;
		}
	}
	
	int len = strlen(str+1);
	for(int i = len; i >= 1; i--)
	{
		int tmp = str[i]-'0';
		if(tmp > 0)
		{
			mat t = qpow(x,tmp);
			
			e = mul(e,t);
		}
		x = qpow(x,10);
	}
	
	return e;
}

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0),cout.tie(0);
	
	cin >> x0 >> x1 >> a >> b >> str+1 >> mod;
	
	mat res = qpow10();
	
	//cout << (res.m[0][0]%mod) << endl;
	
	cout <<  ( (res.m[1][0]*x1%mod)+(res.m[1][1]*x0%mod) )%mod << endl;
	
	
	return 0;	
}