JZOJ 3366. 【NOI2012】随机数生成器

Description

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

$X_{n+1}=(a*X_n+c) mod m$

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

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

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

Input

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

Output

输出一个数，即Xn mod g

Sample Input

11 8 7 1 5 3

Sample Output

2

Data Constraint

Solution

• 这题是典型的矩阵乘法！

• 观察题目，目标是将 $X_0$ 转移成 $X_n$ 且复杂度必须为 $log$ 级别。

• 敏锐地考虑矩阵乘法——（ $X_{i-1} ， c$ ）—》（ $X_i，c$ ）

• 那么友矩阵显然就是

  $$\left[ \begin{matrix} a & 0 \\ 1 & 1 \end{matrix} \right]$$

• 矩阵快速幂转移即可！

• 还要注意中间有 $long long$ 相乘取模，需要使用技巧如下来避免高精度

• 对于两个数 $a*b$ ，以 $10^9$ 为分界隔开，相当于变成 $(a1*10^9+a2)*(b1*10^9+b2)$

• 拆开可得： $a1*b1*10^{18}+a1*b2*10^9+a2*b1*10^9+a2*b2$

• 对于 $10^x$ ，只需进行 x <script type="math/tex" id="MathJax-Element-2034">x</script> 次乘 2 乘 5 即可，之后直接相乘取模！

Code

#include<iostream>
#include<cstring>
using namespace std;
typedef long long LL;
const LL N=1e9;
LL m,n,g;
LL a[3],b[3],c[3][3],d[3][3],f[3][3];
LL get(LL x,LL y)
{
    LL a1=x/N,a2=x%N,b1=y/N,b2=y%N;
    LL t=a1*b1%m,t1=a1*b2%m,t2=a2*b1%m;
    for(int i=1;i<=18;i++) t=t*2%m*5%m;
    t=(t+a2*b2%m)%m;
    for(int i=1;i<=9;i++) t1=t1*2%m*5%m;
    for(int i=1;i<=9;i++) t2=t2*2%m*5%m;
    t=(t+t1)%m;
    t=(t+t2)%m;
    return t;
}
void ksm(LL x)
{
    memcpy(f,c,sizeof(f));
    while(x)
    {
        if(x&1)
        {
            memset(d,0,sizeof(d));
            for(int i=1;i<=2;i++)
                for(int j=1;j<=2;j++)
                    for(int k=1;k<=2;k++)
                        d[i][j]=(d[i][j]+get(f[i][k],c[k][j]))%m;
            memcpy(f,d,sizeof(f));
        }
        memset(d,0,sizeof(d));
        for(int i=1;i<=2;i++)
            for(int j=1;j<=2;j++)
                for(int k=1;k<=2;k++)
                    d[i][j]=(d[i][j]+get(c[i][k],c[k][j]))%m;
        memcpy(c,d,sizeof(c));
        x>>=1;
    }
}
int main()
{
    cin>>m>>c[1][1]>>a[2]>>a[1]>>n>>g;
    c[2][1]=c[2][2]=1;
    ksm(n-1);
    for(int i=1;i<=2;i++)
        for(int j=1;j<=2;j++)
            b[i]=(b[i]+get(a[j],f[j][i]))%m;
    cout<<b[1]%g;
    return 0;
}