[日常摸鱼]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;
}