[卷积定理] LOJ#548. 「LibreOJ β Round #7」某少女附中的体育课

设变换矩阵为 T T

由卷积定理可以知道

对于 T T 的每一行的任意 i,j i , j 满足 xi×xj=xi opt xj x i × x j = x i   opt   x j

因为 A A 满足循环律

所以存在 c>1 c > 1 满足 xci=xi x i c = x i 也就是说 xi x i c c 次单位根

而模数有1到22次的单位根,所以可以爆搜变换矩阵

逆变换只要把变换矩阵求个逆就可以了

然后FWT搞搞

#include 
#include 
#include 
#include 
#include 
#define fi first
#define se second

using namespace std;

typedef long long ll;

typedef pair<int,int> pii;

const int P=232792561;
const int N=25;

int n,m,t,w[N];
ll k;

inline int Pow(int x,int y){
  int ret=1;
  for(;y;y>>=1,x=1LL*x*x%P) if(y&1) ret=1LL*ret*x%P;
  return ret;
}

int A[N][N],p[500010],order[N];

vector ck1[N],ck2[N];

int trans[2][N][N],cnt,df[N];

void dfs(int x){
  if(x>=m){
    int flg=0;
    for(int i=0;i0][cnt][i]=df[i],flg|=df[i];
    if(flg) cnt++;
    return ;
  }
  if(cnt==m) return ;
  int omega=w[order[x]-1],tmp=1;
  for(int i=0;i1LL*tmp*omega%P; df[x]=tmp;
    if(!i) df[x]=0;
    int flg=1;
    for(pii u : ck1[x])
      if(1LL*df[u.fi]*df[u.se]%P!=df[x]){ flg=0; break; }
    if(!flg) continue;
    for(pii u : ck2[x])
      if(1LL*df[u.fi]*df[x]%P!=df[u.se]){ flg=0; break; }
    if(!flg) continue;
    dfs(x+1);
  }
}

inline void Inv(int (*a)[N],int (*b)[N]){
  static int tmp[N][N];
  memcpy(tmp,a,sizeof(tmp));
  for(int i=0;i1;
  for(int i=0;iint k=i; for(;!a[k][i];k++);
    if(k^i)
      for(int j=0;jint t=Pow(a[i][i],P-2);
    for(int j=0;j1LL*a[i][j]*t%P,b[i][j]=1LL*b[i][j]*t%P;
    for(int j=0;jif(j==i || !a[j][i]) continue;
      int t=a[j][i];
      for(int k=0;k1LL*t*a[i][k]%P)%P;
    b[j][k]=(b[j][k]+P-1LL*t*b[i][k]%P)%P;
      }
    }
  }
  memcpy(a,tmp,sizeof(tmp));
}

void FWT(int l,int r,int f){
  if(l+1==r) return ;
  int d=(r-l)/m;
  for(int i=0;i1)*d,f);
  static int tmp[N];
  for(int i=0;ifor(int j=0;jfor(int j=0;jint &cur=p[l+j*d+i]; cur=0;
      for(int k=0;k1LL*trans[f][j][k]*tmp[k])%P;
    }
  }
}

int main(){
  freopen("1.in","r",stdin);
  freopen("1.out","w",stdout);
  for(int i=1;i<=22;i++) w[i]=Pow(71,(P-1)/i);
  scanf("%d%d%lld",&n,&m,&k); t=Pow(m,n);
  for(int i=0;ifor(int j=0;jscanf("%d",&A[i][j]);
  for(int i=0;iscanf("%d",&p[i]);
  for(int i=0;i2;
    for(int j=A[i][i];j!=i;j=A[j][i],order[i]++);
  }
  for(int i=0;ifor(int j=i;jif(A[i][j]>=j) ck1[A[i][j]].push_back(pii(i,j));
      else ck2[j].push_back(pii(i,A[i][j]));
    }
  dfs(0); Inv(trans[0],trans[1]);
  FWT(0,t,0);
  for(int i=0;i1)%(P-1));
  FWT(0,t,1);
  for(int i=0;iprintf("%d\n",p[i]);
  return 0;
}

你可能感兴趣的:(FFT,&,FWT,矩阵)