Description
栋栋最近迷上了随机算法,而随机数生成是随机算法的基础。栋栋准备使用线性同余法(Linear Congruential Method)来生成一个随机数列,这种方法需要设置四个非负整数参数 m, a, c, X0,按照下面的公式生成出一系列随机数:
Xn+1=(aXn+c) mod m
其中 mod m 表示前面的数除以m的余数。从这个式子可以看出,这个序列的下一个数总是由上一个数生成的。
用这种方法生成的序列具有随机序列的性质,因此这种方法被广泛地使用,包括常用的 C++和 Pascal 的产生随机数的库函数使用的也是这种方法。
栋栋知道这样产生的序列具有良好的随机性,不过心急的他仍然想尽快知道Xn是多少。由于栋栋需要的随机数是0, 1, … ,n−1 之间的,他需要将 Xn 除以g取余得到他想要的数,即 Xn mod g,你只需要告诉栋栋他想要的数Xn mod g是多少就可以了。
Solution
多么裸的矩阵乘法啊!
复杂度:
O(log(n))
副:LL*LL取模
long long cm(long long a, long long b)
{
LL c=mo;
long long ret = 0;
while(b)
{
if(b & 1)
{
ret += a;
if(ret >= c)
ret -= c;
}
a += a;
if(a >= c) a -= c;
b >>= 1;
}
return ret;
}
Code
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
typedef long long LL;
LL n,m,ans;
LL mo,a1,c,x0,mg;
LL a[2],b[2][2];
long long cm(long long a, long long b)
{
LL c=mo;
long long ret = 0;
while(b)
{
if(b & 1)
{
ret += a;
if(ret >= c)
ret -= c;
}
a += a;
if(a >= c) a -= c;
b >>= 1;
}
return ret;
}
void chenans()
{
a[0]=(cm(a[0],b[0][0])+b[1][0])%mo;
if(b[0][0]<0)printf("s\n");
}
void chen()
{
LL s[2][2]={0};
fo(i,0,1)
fo(j,0,1)
fo(k,0,1)
{
s[i][j]=(s[i][j]+cm(b[i][k],b[k][j]))%mo;
if(s[i][j]<0)printf("%lld\n",cm(b[i][k],b[k][j]));
}
fo(i,0,1)fo(j,0,1)b[i][j]=s[i][j];
}
int main()
{
int q,w;
scanf("%lld%lld%lld%lld%lld%lld",&mo,&a1,&c,&x0,&n,&mg);
b[0][0]=a1,b[1][0]=c,b[1][1]=1;
a[0]=x0,a[1]=1;
while(n)
{
if(n&1)chenans();
chen();
n>>=1;
}
printf("%lld\n",a[0]%mg);
return 0;
}