「HNOI 2019」白兔之舞

一道清真的数论题

LOJ #3058

Luogu P5293

---

题解

考虑$ n=1$的时候怎么做

设$ s$为转移的方案数

设答案多项式为$\sum\limits_{i=0}^L (sx)^i\binom{L}{i}=(sx+1)^L$

答案相当于这个多项式模$ k$的各项系数的和

发现这和LJJ学二项式定理几乎一模一样

我上一题的题解

然而直接搞是$ k^2$的，无法直接通过本题

以下都用$ w$表示$ k$次单位根

设$ F_i$为次数模$ k$为$ i$的项的系数和

单位根反演一下得到$F(i)=\sum\limits_{j=0}^{k-1}w^{-ij}(sw^j+1)^L$

组合数有个非常优美的性质

$$ij=\binom{i+j}{2}-\binom{i}{2}-\binom{j}{2}$$

推波式子

$$
\begin{aligned}
F(i)&=\sum_{j=0}^{k-1}w^{-ij}(sw^j+1)^L\\
&=\sum_{j=0}^{k-1}w^{\binom{i}{2}+\binom{j}{2}-\binom{i+j}{2}}(sw^j+1)^L\\
&=w^\binom{i}{2}\sum_{j=0}^{k-1}w^{-\binom{i+j}{2}}w^\binom{j}{2}(sw^j+1)^L\\
\end{aligned}
$$

发现这是一个差卷积的形式

扔一个$ MTT$上去就能拿那$ 40$分了

考虑$ n>1$的情况

相当于把$ s$从一个数变成了矩阵，把$ 1$变成单位矩阵

$ (sw^j+1)^L$这个矩阵我们只需要关注一个位置上的值

因此可以乘出来然后取[x][y]这个位置上的数即可

这样就可以通过此题

复杂度：大常数单 $\log$

---

代码

#include<ctime>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#include<vector>
#define rt register int
#define ll long long
using namespace std;
inline ll read(){
    ll x=;char zf=;char ch=getchar();
    while(ch!='-'&&!isdigit(ch))ch=getchar();
    if(ch=='-')zf=-,ch=getchar();
    while(isdigit(ch))x=x*+ch-'',ch=getchar();return x*zf;
}
void write(ll y){if(y<)putchar('-'),y=-y;if(y>)write(y/);putchar(y%+);}
void writeln(const ll y){write(y);putchar('\n');}
int k,m,n,x,y,cnt,p,L;
int w[],val[];
int f[],g[],F[];
int ksm(int x,int y=p-){
    int ans=;
    for(;y;y>>=,x=1ll*x*x%p)if(y&)ans=1ll*ans*x%p;
    return ans;
}
struct mat{
    ll a[][];
    void print(){
        for(rt i=;i<n;i++)for(rt j=;j<n;j++)cout<<a[i][j]<<" \n"[j==n-];
    }
    inline mat operator*(const mat s)const{
        mat ret={};
        for(rt i=;i<n;i++)
        for(rt k=;k<n;k++)
        for(rt j=;j<n;j++)
        (ret.a[i][j]+=a[i][k]*s.a[k][j]);
        for(rt i=;i<n;i++)for(rt j=;j<n;j++)ret.a[i][j]%=p;
        return ret;
    }
    mat operator*(const int s)const{
        mat ret;
        for(rt i=;i<n;i++)
        for(rt j=;j<n;j++)
        ret.a[i][j]=a[i][j]*s%p;
        return ret;
    }
    mat operator+(const mat s)const{
        mat ret;memset(ret.a,,sizeof(ret.a));
        for(rt i=;i<n;i++)
        for(rt j=;j<n;j++)
        ret.a[i][j]=(a[i][j]+s.a[i][j])%p;
        return ret;
    }

}I,zy;
mat ksm(mat x,int y){
    mat ans=I;
    for(;y;y>>=,x=x*x)if(y&)ans=ans*x;
    return ans;
}
int V(int x){
    return 1ll*x*(x-)/%k;
}
const double PI=acos(-1.0);
struct cp{
    double x,y;
    cp operator +(const cp &s)const{return {x+s.x,y+s.y};}
    cp operator -(const cp &s)const{return {x-s.x,y-s.y};}
    cp operator *(const cp &s)const{return {x*s.x-y*s.y,x*s.y+y*s.x};}
}W[][<<],A[],B[],C[],D[];
int R[];
void FFT(const int n,cp *A){
    for(rt i=;i<n;i++)if(i>R[i])swap(A[i],A[R[i]]);
    for(rt j=;j<n;j+=){
        const cp x=A[j],y=A[j+];
        A[j]=x+y,A[j+]=x-y;
    }
    for(rt i=,s=;i<n;i<<=,s++)
    for(rt j=;j<n;j+=i<<)
    for(rt k=;k<i;k+=){
        cp x=A[j+k],y=W[s][k]*A[i+j+k];
        A[j+k]=x+y;A[i+j+k]=x-y;
        x=A[j+k+],y=W[s][k+]*A[i+j+k+];
        A[j+k+]=x+y;A[i+j+k+]=x-y;
    }
}
int yg(int n){
    for(rt i=;i<=n;i++){
        for(rt j=;j*j<=n;j++)if((n-)%j==)
        if(ksm(i,(n-)/j)==)goto end;
        return i;end:;
    }
}
void subtask(){
    w[]=;w[]=ksm(yg(p),(p-)/k);
    for(rt i=;i<k;i++)w[i]=1ll*w[i-]*w[]%p;mat s=zy;
    for(rt i=;i<k;i++){
        g[i]=ksm(s*w[i]+I,L).a[x-][y-]*w[V(i)]%p;
    }
    for(rt i=;i<k+k;i++)f[i]=w[(k-V(i))%k];
    for(rt i=;i<k/;i++)swap(g[i],g[k-i]);

    n=k+k-;m=k;
    int lim=;while(lim<=n+m)lim<<=;
    for(rt i=;(<<i)<lim;i++){
        W[i][]={,};
        W[i][]={cos(PI/(<<i)),sin(PI/(<<i))};
        for(rt j=;j<<<i;j++)
        if(j%==)W[i][j]={cos(PI*j/(<<i)),sin(PI*j/(<<i))};
        else W[i][j]=W[i][j-]*W[i][];
    }

    for(rt i=;i<=n;i++){
        A[i].x=f[i]>>;
        A[i].y=f[i]&;
    }
    for(rt i=;i<=m;i++){
        B[i].x=g[i]>>;
        B[i].y=g[i]&;
    }
    for(rt i=;i<lim;i++)R[i]=(R[i>>]>>)|(i&)*(lim>>);
    FFT(lim,A);FFT(lim,B);
    for(rt i=;i<lim;i++){
        const int pl=(lim-)&(lim-i);
        const cp ca={A[pl].x,-A[pl].y},cb={B[pl].x,-B[pl].y};
        const cp a=(A[i]+ca)*(cp){0.5,},b=(A[i]-ca)*(cp){,-0.5},
                  c=(B[i]+cb)*(cp){0.5,},d=(B[i]-cb)*(cp){,-0.5};
        C[pl]=a*c+a*d*(cp){,};D[pl]=b*c+b*d*(cp){,};
    }
    FFT(lim,C);FFT(lim,D);
    for(rt i=k;i<k+k;i++){
        const int vv=1ll*w[V(i-k)]*ksm(k,p-)%p;
        ll a=C[i].x/lim+0.5,b=C[i].y/lim+0.5,c=D[i].x/lim+0.5,d=D[i].y/lim+0.5;
        a=((a%p)<<)+(((b+c)%p)<<)+d;a=(a%p+p)%p;
        writeln(1ll*a*vv%p);
    }
    exit();
}
int jc[],njc[],inv[],ff[][],ans[];
int c(int x,int y){
    return 1ll*jc[x]*njc[y]%p*njc[x-y]%p;
}
void bl(){
    for(rt i=;i<;i++)jc[i]=njc[i]=inv[i]=;
    for(rt i=;i<=L;i++){
        jc[i]=1ll*jc[i-]*i%p;
        inv[i]=1ll*inv[p%i]*(p-p/i)%p;
        njc[i]=1ll*njc[i-]*inv[i]%p;
    }
    ff[][x-]=;if(x==y)ans[]=;
    for(rt i=;i<=L;i++)
    for(rt j=;j<n;j++){
        for(rt k=;k<n;k++)
        (ff[i][j]+=1ll*zy.a[k][j]*ff[i-][k]%p)%=p;
        if(j==y-)(ans[i%k]+=1ll*ff[i][j]*c(L,i)%p)%=p;
    }
    for(rt i=;i<k;i++)writeln(ans[i]);exit();
}
int main(){
    cin>>n>>k>>L>>x>>y>>p;
    for(rt i=;i<n;i++)
    for(rt j=;j<n;j++)
    cin>>zy.a[i][j];
    if(L<=)bl();
    for(rt i=;i<n;i++)I.a[i][i]=;
    subtask();
    return ;
}