其他分享
首页 > 其他分享> > [HNOI2019]白兔之舞(矩阵快速幂+单位根反演)

[HNOI2019]白兔之舞(矩阵快速幂+单位根反演)

作者:互联网

非常抱歉,这篇文章鸽了。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define cp complex<long double>
#define pi acosl(-1)
const int N=6e5+7;
struct mat{ll a[5][5];}A1,A2;
ll n,k,l,X,Y,mod,w,nn,ans[N],a1[N],a2[N],R[N];
cp A[N],B[N],C[N],D[N],E[N],F[N],G[N];
ll qpow(ll a,ll b)
{
    ll ret=1;
    while(b)
    {
        if(b&1)ret=ret*a%mod;
        a=a*a%mod,b>>=1;
    }
    return ret;
}
mat operator*(mat a,mat b)
{
    mat c;
    for(int i=1;i<=n;i++)
    for(int j=1;j<=n;j++)
    {
        c.a[i][j]=0;
        for(int k=1;k<=n;k++)c.a[i][j]=(c.a[i][j]+a.a[i][k]*b.a[k][j])%mod;
    }
    return c;
}
mat qpow(mat a,ll b)
{
    mat ret={0};
    for(int i=1;i<=n;i++)
    for(int j=1;j<=n;j++)
    if(i==j)ret.a[i][j]=1;
    while(b)
    {
        if(b&1)ret=ret*a;
        a=a*a,b>>=1;
    }
    return ret;
}
ll getrt(ll x)
{
    vector<int>vec;vec.clear();
    for(ll i=2;i*i<=(x-1);i++)if((x-1)%i==0)vec.push_back(i),vec.push_back((x-1)/i);
    for(ll i=2;;i++)
    {
        bool c=1;
        for(unsigned j=0;j<vec.size()&&c;j++)if(qpow(i,vec[j])==1)c=0;
        if(c)return i;
    }
}
ll c1(ll x){return x*(x-1)/2;}
void FFT(cp*a,int f)
{
    for(int i=0;i<nn;i++)if(i<R[i])swap(a[i],a[R[i]]);
    for(int i=1;i<nn;i*=2)
    {
        cp wn(cosl(pi/i),sinl(pi/i));
        for(int j=0;j<nn;j+=i*2)
        {
            cp w(1,0),x,y;
            for(int k=0;k<i;k++,w=w*wn)
            x=a[j+k],y=a[j+k+i]*w,a[j+k]=x+y,a[j+k+i]=x-y;
        }
    }
    if(f==1)return;
    reverse(a+1,a+nn);
    for(int i=0;i<nn;i++)a[i]/=nn;
}
int main()
{
    cin>>n>>k>>l>>X>>Y>>mod;
    for(int i=1;i<=n;i++)for(int j=1;j<=n;j++)cin>>A1.a[i][j];
    w=qpow(getrt(mod),(mod-1)/k);
    for(int j=0;j<k;j++)
    {
        for(int i=1;i<=n;i++)
        for(int p=1;p<=n;p++)
        {
            A2.a[i][p]=A1.a[i][p]*qpow(w,j)%mod;
            if(i==p)A2.a[i][p]++;
        }
        A2=qpow(A2,l),a2[j]=A2.a[X][Y];
    }
    for(int j=0;j<k;j++)a2[j]=a2[j]*qpow(w,c1(j))%mod;
    for(int j=0;j<=k*2;j++)a1[j]=qpow(w,c1(j)*(mod-2));
    reverse(a1,a1+k*2+1);
    int m=k*3,L=0;
    for(int i=0;i<m;i++)A[i]=a2[i]/32768,B[i]=a2[i]%32768;
    for(int i=0;i<m;i++)C[i]=a1[i]/32768,D[i]=a1[i]%32768;
    for(nn=1;nn<=m;nn<<=1)L++;
    for(int i=0;i<nn;i++)R[i]=(R[i>>1]>>1)|((i&1)<<L-1);
    FFT(A,1),FFT(B,1),FFT(C,1),FFT(D,1);
    for(int i=0;i<nn;i++)E[i]=A[i]*C[i],F[i]=A[i]*D[i]+B[i]*C[i],G[i]=B[i]*D[i];
    FFT(E,-1),FFT(F,-1),FFT(G,-1);
    for(int i=0;i<nn;i++)
    ans[i]=((ll)(G[i].real()+0.5)+(((ll)(F[i].real()+0.5)%mod)<<15)+((ll)(E[i].real()+0.5)%mod<<30))%mod;
    for(ll i=0;i<k;i++)
    {
        ans[k*2-i]=(ans[k*2-i]*qpow(k,mod-2)%mod)*qpow(w,c1(i))%mod;
        printf("%lld\n",ans[k*2-i]);
    }
}
View Code

 

标签:return,mat,int,ll,单位根,ret,反演,HNOI2019,mod
来源: https://www.cnblogs.com/hfctf0210/p/10952262.html