[日常摸鱼]bzoj2875[NOI2012]随机数生成器-矩阵快速幂
好裸的矩阵快速幂…然而我一开始居然构造不出矩阵…
平常两个的情况都是拿相邻两项放在矩阵里拿去递推的…然后我就一直构造不出来…其实把矩阵下面弄成1就好了啊orz
#include<cstdio> #include<cstring> #define rep(i,n) for(register lint i=1;i<=n;i++) typedef long long lint; lint m,n,g; struct matrix { lint m[3][3]; matrix(){memset(m,0,sizeof(m));} }; inline lint mul(lint a,lint b) { lint res=0; for(;b;b>>=1,a=(a+a)%m)if(b&1)res=(res+a)%m; return res; } inline matrix operator *(matrix a,matrix b) { matrix res; rep(i,2)rep(j,2)rep(k,2)res.m[i][j]=(res.m[i][j]+mul(a.m[i][k],b.m[k][j]))%m; return res; } inline matrix pow_mod(matrix a,lint b) { matrix res;res.m[1][1]=res.m[2][2]=1; for(;b;b>>=1,a=a*a)if(b&1)res=a*res; return res; } int main() { matrix t,r; scanf("%lld%lld%lld%lld%lld%lld",&m,&t.m[1][1],&t.m[1][2],&r.m[1][1],&n,&g); r.m[2][1]=t.m[2][2]=1;r=pow_mod(t,n)*r;printf("%lld",r.m[1][1]%g); return 0; }