BZOJ1547 周末晚会题解（置换群+数论+计数DP）

题目：BZOJ1547.
题目大意：给定一个长度为$n$的环，要求每个位置染成$0$或$1$，其中连续的$0$数量不超过$m$，求方案数.循环同构算一种方案但翻转同构不算.
数据组数$T\le 50$，$1\le n,m\le 2000$，答案对$10^8+7$取模.

首先看到了循环同构算一种方案就可以想到用Burnside引理转化成统计$n$个置换中的每个循环染同色的方案数.

考虑对于一个转了$a$次的循环，根据一些数论知识，我们知道总共会有$gcd(n,a)$个循环，显然这些循环会形成一些长度为$\frac{n}{gcd(n,a)}$的循环节，每个循环节内部的所有位置互不相干.

问题变成了如何统计一个长度为$gcd(n,a)$的环中没有超过$m$个连续的$0$的染色方案.显然若$m+1\ge gcd(n,a)$时答案就是$2^{gcd(n,a)}-[m+1\le n]$，现在考虑$m\le gcd(n,a)$的情况.

发现此时环并不好解决但是链好办，所以先统计链的方案数，再减去在链上没有超过$m$个$0$但首尾相连后就有的方案数即可.

为了实现这一点，考虑一个DP.设$f[i][j]$表示长度为$i$的链上满足连续的$0$不超过$m$个且倒数第$j+1$个染为$1$最后$j$个都染为$0$的方案数，容易得到状态转移方程：
$$f[i][j]=\{\begin{array}{cc}{\sum }_{k=0}^{m}f[i-1][k]& j=0\\ f[i-1][j-1]& j>0\end{array}$$

链的方案数显然为${\sum }_{i=0}^{m}f[gcd(n,a)][i]$，然后考虑计算链上没有超过$m$个连续的$0$但首尾相连后有的方案数.

考虑枚举开头连续$0$的数量$i$，再枚举结尾连续$0$的数量，就可以得到方案数为：
$$\sum _{i=1}^{\min (gcd(n,a)-1,m)}\sum _{j=m+1-i}^{m}f[gcd(n,a)][j]$$

前缀和一下就可以省去一个$\sum$，时间复杂度降为$O(m)$.

总时间复杂度$O(Tnm)$.

代码如下：

#include<bits/stdc++.h>
using namespace std;

#define Abigail inline void
typedef long long LL;

const int N=2000,mod=100000007;

int add(int a,int b){return a+b>=mod?a+b-mod:a+b;}
int sub(int a,int b){return a-b<0?a-b+mod:a-b;}
int mul(int a,int b){return (LL)a*b%mod;}
void sadd(int &a,int b){a=add(a,b);}
void ssub(int &a,int b){a=sub(a,b);}
void smul(int &a,int b){a=mul(a,b);}
int Power(int a,int k){int res=1;for (;k;k>>=1,smul(a,a)) if (k&1) smul(res,a);return res;}
int Get_inv(int a){return Power(a,mod-2);}
int gcd(int a,int b){return b?gcd(b,a%b):a;}

int n,m;
int dp[N+9][N+9];

void Get_dp(){
  dp[0][0]=1;
  for (int i=1;i<=n;++i){
    dp[i][0]=dp[i][m+1]=0;
  	for (int j=0;j<=m;++j){
  	  sadd(dp[i][0],dp[i-1][j]);
      if (j>0) dp[i][j]=dp[i-1][j-1];
	}
  }
  for (int i=1;i<=n;++i)
    for (int j=m;j>=0;--j) sadd(dp[i][j],dp[i][j+1]);
}

int Solve(int len){
  if (len<=m+1) return sub(Power(2,len),m+1<=n);
  int res=dp[len][0];
  for (int i=1;i<len&&i<=m;++i) ssub(res,dp[len-i-1][m+1-i]);
  return res;
}

int ans;

Abigail into(){
  scanf("%d%d",&n,&m);
}

Abigail work(){
  Get_dp();
  ans=0;
  for (int i=0;i<n;++i) sadd(ans,Solve(gcd(n,i)));
}

Abigail outo(){
  printf("%d\n",mul(ans,Get_inv(n)));
}

int main(){
  int T;
  scanf("%d",&T);
  while (T--){
    into();
    work();
    outo();
  }
  return 0;
}