【矩阵快速幂】[NOI2012]随机数生成器

题目描述

栋栋最近迷上了随机算法,而随机数生成是随机算法的基础。栋栋准备使用线性同余法(Linear Congruential Method)来生成一个随机数列,这种方法需要设置四个非负整数参数 m,a,c,X0,按照下面的公式生成出一系列随机数<Xn>: 
Xn+1=(aXn+c) MOD m;
其中 MOD m 表示前面的数除以 m 的余数。从这个式子可以看出,这个序列的下一个数总是由上一个数生成的。 

用这种方法生成的序列具有随机序列的性质,因此这种方法被广泛地使用,包括常用的C++和Pascal的产生随机数的库函数使用的也是这种方法。 栋栋知道这样产生的序列具有良好的随机性,不过心急的他仍然想尽快知道Xn是多少。由于栋栋需要的随机数是0,1, …, g-1 之间的,他需要将 Xn除以 g 取余得到他想要的数,即 Xn MOD g,你只需要告诉栋栋他想要的数Xn MOD g是多少就可以了。 

输入

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

输出

输出Xn MOD g

样例输入

Copy

(如果复制到控制台无换行,可以先粘贴到文本编辑器,再复制)

11 8 7 1 5 3

样例输出


2

提示


n<=10^18

分析

就是裸的一矩阵快速幂,需要用到模乘法。

代码

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
#define MOD m
long long m,a,c,x,n,g;
void Read(long long &x){
    static char c;
    while(c=getchar(),c!=EOF)
        if(c>='0'&&c<='9'){
            x=c-'0';
            while(c=getchar(),c>='0'&&c<='9')
                x=x*10+c-'0';
            ungetc(c,stdin);
            return;
        }
}
long long quick_mul(long long a,long long b){
    long long ret(0);
    while(b){
        if(b&1)
            ret=(ret+a)%MOD;
        a=(a+a)%MOD;
        b>>=1;
    }
    return ret;
}
struct Matrix{
    long long a[2][2];
    inline Matrix(){
        memset(a,0,sizeof a);
    }
    inline Matrix(int){
        memset(a,0,sizeof a);
        for(int i=0;i<2;i++)
            a[i][i]=1;
    }
    inline Matrix operator*(const Matrix &b){
        Matrix c;
        int i,j,k;
        for(i=0;i<2;i++)
            for(j=0;j<2;j++)
                if(a[i][j])
                    for(k=0;k<2;k++)
                        c.a[i][k]=(c.a[i][k]+quick_mul(a[i][j],b.a[j][k]))%MOD;
        return c;
    }
    inline Matrix& operator*=(const Matrix&b){
        return *this=*this*b;
    }
}A;
Matrix quick_pow(Matrix a,long long b){
    Matrix ret(1);
    while(b){
        if(b&1)
            ret*=a;
        a*=a;
        b>>=1;
    }
    return ret;
}
void read(){
    Read(m),Read(a),Read(c),Read(x),Read(n),Read(g);
}
void solve(){
    A.a[0][0]=a,A.a[1][0]=c,A.a[1][1]=1;
    A=quick_pow(A,n);
}
int main()
{
    read();
    solve();
    printf("%lld\n",(quick_mul(x,A.a[0][0])+A.a[1][0])%MOD%g);
}
posted @ 2016-07-13 10:44  outer_form  阅读(144)  评论(0编辑  收藏  举报