【CODEVS1281】Xn数列

Description

给你6个数，m, a, c, x0, n, g

Xn+1 = ( aXn + c ) mod m，求Xn

m, a, c, x0, n, g<=10^18

Input

一行六个数 m, a, c, x0, n, g

Output

输出一个数 Xn mod g

Sample Input

11 8 7 1 5 3

Sample Output

2

HINT

int64按位相乘可以不要用高精度。

题解

对于取模运算进行一个修改，将乘法修改为倍增乘法(类似于快速幂)：

long long Multiply (long long x, long long y)
{
long long ans = 0;
while(y)
{
if(y & 1) ans = (ans + x) % m;
x = (x << 1) % m;
y >>= 1;
}
return ans;
}

代码粘黄学长的

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
long long m,a,c,x0,n,g;
long long Multiply (long long x,long long y)
{
long long ans=0;
while (y)
{
if (y&1) ans=(ans+x)%m;
x=(x<<1)%m;
y>>=1;
}
return ans;
}
void mul(long long a[2][2],long long b[2][2])
{
long long c[2][2]={{0,0},{0,0}};
for (int i=0;i<2;i++)
for (int j=0;j<2;j++)
for (int k=0;k<2;k++)
c[i][j] = (c[i][j]+Multiply(a[i][k],b[k][j]))%m;//就是这里
memcpy(a,c,sizeof(c));
}
int main()
{
cin>>m>>a>>c>>x0>>n>>g;
long long f[2][2] = {{x0,1},{0,0}};
long long b[2][2] = {{a,0},{c,1}};
while(n)
{
if (n&1) mul(f,b);
mul(b,b);
n >>= 1;
}
cout<<f[0][0]%g;
}