NOI2012 随机数生成器

题目链接:戳我

\((x[0],1)\)乘上
\( \begin{pmatrix} a & 0\\ c & 1\\ \end{pmatrix} \)
的n次方就行了

记得乘法可能炸long long,所以要快速乘qwq

#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<cstdio>
#define ll long long
using namespace std;
ll m,a,c,g,n,x0;
struct Node{ll t[2][2];}init;
inline ll mul(ll x,ll y,ll mod)
{
    ll cur_ans=0;
    while(y)
    {
        if(y&1) cur_ans=(cur_ans+x)%mod;
        x=(x+x)%mod;
        y>>=1;
    }
    return cur_ans;
}
inline Node calc(Node x,Node y,ll mod)
{
    Node cur;
    for(int i=0;i<=1;i++)
        for(int j=0;j<=1;j++)
            cur.t[i][j]=0;
    for(int i=0;i<=1;i++)
        for(int j=0;j<=1;j++)
            for(int k=0;k<=1;k++)
              cur.t[i][j]=(cur.t[i][j]+mul(x.t[i][k],y.t[k][j],mod))%mod;
    return cur;
}
inline Node fpow(Node x,ll y,ll mod)
{
    Node cur;
    for(int i=0;i<=1;i++)
        for(int j=0;j<=1;j++)
            cur.t[i][j]=0;
    for(int i=0;i<=1;i++) cur.t[i][i]=1;
    while(y)
    {
        if(y&1) cur=calc(cur,x,mod);
        x=calc(x,x,mod);
        y>>=1;
    }
    return cur;
}
int main()
{
    #ifndef ONLINE_JUDGE
    freopen("ce.in","r",stdin);
    #endif
    scanf("%lld%lld%lld%lld%lld%lld",&m,&a,&c,&x0,&n,&g);
    init.t[0][0]=a;
    init.t[0][1]=0;
    init.t[1][0]=c;
    init.t[1][1]=1;
    // puts("");
    init=fpow(init,n,m);
    ll cur_ans=mul(x0,init.t[0][0],m);
    // cout<<cur_ans<<endl;
    cur_ans=(cur_ans+init.t[1][0])%m;
    cur_ans%=g;
    printf("%lld\n",cur_ans);
    return 0;
}