【Bzoj2875】随机数生成器

题目描述

栋栋最近迷上了随机算法，而随机数是生成随机算法的基础。栋栋准备使用线性同余法（Linear Congruential Method）来生成一个随机数列，这种方法需要设置四个非负整数参数m,a,c,X[0],按照下面的公式生成出一系列随机数{Xn}：

X[n+1]=(aX[n]+c) mod m

其中mod m表示前面的数除以m的余数。从这个式子可以看出，这个序列的下一个数总是由上一个数生成的。

用这种方法生成的序列具有随机序列的性质，因此这种方法被广泛地使用，包括常用的C++和Pascal的产生随机数的库函数使用的也是这种方法。

栋栋知道这样产生的序列具有良好的随机性，不过心急的他仍然想尽快知道X[n]是多少。由于栋栋需要的随机数是0,1,…,g-1之间的，他需要将X[n]除以g取余得到他想要的数，即X[n] mod g，你只需要告诉栋栋他想要的数X[n] mod g是多少就可以了。

输入输出格式

输入格式：
输入包含6个用空格分割的整数m,a,c,X[0],n和g，其中a,c,X[0]是非负整数，m,n,g是正整数。

输出格式：
输出一个数，即X[n] mod g

输入输出样例

输入样例#1：
11 8 7 1 5 3
输出样例#1：
2
说明

计算得X[n]=X[5]=8,故(X[n] mod g) = (8 mod 3) = 2

100%的数据中n,m,a,c,X[0]<=10^18,g<=10^8

---

算法

$矩阵乘法，我们发现由X[n+1]=(aX[n]+c) mod m可以将X[n]表示为a^nx_0+c(a^{n-1}+a^{n-2}+...+a+1)$
$构造出一个形如 \left[ \begin{matrix} a & c\\ 0 & 1 \end{matrix} \right] 的矩阵。之后将它自乘n次，则mar[1][1]=a^n,mar[1][2]=后面的常数部分。$
$注意此题要开LL，还要打快速乘，毕竟NOI题目还是有坑点。$

#include <cstdio>
#include <algorithm>
#include <cstring>

#define Rep(i,s,t) for(int i=s;i<=t;i++)

using namespace std;

typedef long long LL;

LL n,m,a,c,x0,g,Abcd;

LL quick_mul(LL a,LL b){
    LL ans = 0;
    while(b){
        if(b & 1) ans = (ans+a)%m;
        a = (a+a)%m;
        b >>= 1;
    }
    return ans;
}

namespace martrix_mul{
    struct martrix{
        LL mar[3][3];
    }Ans,mimi;

    martrix mul(martrix a,martrix b){
        martrix tmp;
        memset(tmp.mar,0,sizeof(tmp.mar));
        Rep(i,1,2) Rep(j,1,2) Rep(k,1,2)
            tmp.mar[i][j] = (tmp.mar[i][j] + quick_mul(a.mar[i][k],b.mar[k][j]))%m;
        return tmp;
    }

    martrix power(martrix a,LL n){
        martrix tmp = a,Ans = a;
        n--;
        while(n){
            if(n & 1) Ans = mul(Ans,tmp);
            tmp = mul(tmp,tmp);
            n >>= 1;
        }
        return Ans;
    }
}

int main(){ 
    using namespace martrix_mul;
    scanf("%lld%lld%lld%lld%lld%lld",&m,&a,&c,&x0,&n,&g);
    Ans.mar[1][1]=a;Ans.mar[1][2]=c;Ans.mar[2][1]=0;Ans.mar[2][2]=1;
    mimi = power(Ans,n);
    Abcd = (quick_mul(mimi.mar[1][1],x0) + mimi.mar[1][2]) %m %g;
    printf("%lld",Abcd);
    return 0;
}