【JZOJ 3366】随机数生成器

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;
}