题目描述
栋栋最近迷上了随机算法,而随机数是生成随机算法的基础。栋栋准备使用线性同余法(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)modm可以将X[n]表示为anx0+c(an−1+an−2+...+a+1)
构造出一个形如[a0c1]的矩阵。之后将它自乘n次,则mar[1][1]=an,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;
}