Xn数列(矩阵乘法+快速幂+慢速乘法)
Xn数列
题目描述:
给你6个数,m, a, c, x0, n, g
Xn+1 = ( aXn + c ) mod m,求Xn
m, a, c, x0, n, g<=10^18
输入描述:
一行六个数 m, a, c, x0, n, g
输出描述:
输出一个数 Xn mod g
样例输入:
11 8 7 1 5 3
样例输出:
2
数据范围及提示:
int64按位相乘可以不要用高精度。
思路:
暴力循环求解?NO~NO~NO~
如此大的数据,想到矩阵乘法,利用矩阵加速。
这道题的难度在于如何推出矩阵。
首先根据矩阵性质以及运算规则容易看出这是2*2的矩阵
让X0位于a[1][1]的位置。那么运算之后X1肯定也要位于a[1][1]的位置
因为X1=a*X0+c,所以a[1][2]的位置是c,a[2][1]和a[2][2]对于答案没有影响,所以设为0就可以了,至此,初始矩阵设置完毕。
那么初始矩阵乘以哪一个矩阵才能得到答案呢?重新定义一个矩阵b,容易看出,当b[1][1]=a,b[1][2]=0,b[2][1]=1,b[2][2]=1时,矩阵a*矩阵b才能得到我们想要的答案。
可以看出:
a*b可以得出X1,同样的X1*b得出X2,,即:X2=a*b*b,所以Xn=a*b^n。
因为数据太大,所以使用快速幂加速,long long会乘爆,所以使用慢速乘法。
#include<iostream>
#define lon long long
using namespace std;
lon m,a,c,x0,n,g,s[3][3],ans[3][3];
lon slow_mul(lon x,lon y,lon mod)//慢速乘法
{
lon ans=0;
while(y)
{
if(y&1)
{
y--;
ans=(ans+x)%mod;
}
y>>=1;
x=(x+x)%mod;
}
return ans;
}
void mul(lon p[3][3],lon q[3][3])//快速幂
{
lon tmp[3][3]={0};
for(int i=1;i<=2;i++)
for(int j=1;j<=2;j++)
for(int k=1;k<=2;k++)
tmp[i][j]=(tmp[i][j]+slow_mul(p[i][k],q[k][j],m))%m;
for(int i=1;i<=2;i++)
for(int j=1;j<=2;j++)
p[i][j]=tmp[i][j];
}
void quick_power(lon n)
{
lon tmp[3][3];
for(int i=1;i<=2;i++)
for(int j=1;j<=2;j++)
tmp[i][j]=s[i][j];
while(n)
{
if(n&1)
mul(tmp,s);//因为初始值tmp为s,所以为n-1次幂 ,而不是n次幂。。。。
mul(s,s);
n>>=1;
}
for(int i=1;i<=2;i++)
for(int j=1;j<=2;j++)
s[i][j]=tmp[i][j];
}
int main()
{
cin>>m>>a>>c>>x0>>n>>g;
ans[1][1]=x0,ans[1][2]=c,ans[2][1]=0,ans[2][2]=0;
s[1][1]=a,s[1][2]=0,s[2][1]=1,s[2][2]=1;
quick_power(n-1);//为什么是n-1次幂?
lon tmp[3][3]={0};
for(int i=1;i<=2;i++)
for(int j=1;j<=2;j++)
for(int k=1;k<=2;k++)
tmp[i][j]=(tmp[i][j]+slow_mul(ans[i][k],s[k][j],m))%m;
cout<<tmp[1][1]%g;//输出答案
return 0;
}