Atcoder agc036F

当时改题的时候没有意识到这个奇怪题意给出的性质，结果完全不会。
这个问题本身大致是一个不太可做的完美匹配计数，不过我们注意到对于每个$i$，它能匹配到的$P_i$是一个区间$[L_i,R_i]$，且$L$与$R$均单调不增。
如果所有$i$均有$L_i=0$，那么我们容易对$R_i$排序后$\mathcal{O}(N)$计算。
考虑一个容斥的做法，我们枚举一些$L_i>0$的数，将它们的$R_i$换为$L_i-1$，带上$-1$的权，再跑上面的做法，可惜复杂度是指数级的。
进一步的，我们发现有$L_0<R_{N-1}$，且$\forall i\ge N$有$L_i=0$。那么可以取出所有的$R_i$和$L_i-1$，从小到大排序后存在一个分界点，使得左侧为$L_0-1$到$L_{N-1}-1$与$R_N$到$R_{2N-1}$，右侧为$R_0$到$R_{N-1}$。
这样就可以DP了。考虑枚举选了$k$个$L_i-1$，我们在两侧同时DP，设$F[i][j]$表示考虑了左侧后$i$个数以及这些数在右侧对应的后缀，目前左侧选了$j$个$L_i-1$的方案数，因为我们假设了左侧最终会选$k+N$个数，所以容易转移。
时间复杂度$\mathcal{O}(N^3)$。

#include <bits/stdc++.h>
#define FR first
#define SE second

using namespace std;

typedef long long ll;
typedef pair<int,int> pr;

int MOD;

inline void add(int &x,ll y) {
  x=(x+y)%MOD;
}

inline void sub(int &x,ll y) {
  x=(x-y)%MOD;
}

pr num[505];
int f[2][255];

int main() {
  int n;
  scanf("%d%d",&n,&MOD);
  int cnt=0;
  for(int i=0;i<n;i++) {
  	int r1=0;
  	while (r1*r1+i*i<n*n) r1++;
  	int r2=r1;
  	while (r2<2*n-1&&(r2+1)*(r2+1)+i*i<=4*n*n) r2++;
  	num[++cnt]=pr(r1,r2+1);
  }
  for(int i=n;i<2*n;i++) {
  	int r=0;
  	while (r<2*n-1&&(r+1)*(r+1)+i*i<=4*n*n) r++;
  	num[++cnt]=pr(r+1,0);
  }
  sort(num+1,num+cnt+1);
  int ans=0;
  for(int i=0;i<=n;i++) {
  	int cur=0;
  	memset(f[cur],0,sizeof(f[cur]));
  	f[cur][0]=1;
  	int s1=0,s2=0;
  	for(int j=1;j<=cnt;j++) {
  		cur^=1;
  		memset(f[cur],0,sizeof(f[cur]));
  		if (!num[j].SE) {
  			int u=num[j].FR-s2,t=min(i,s1);
  			for(int k=0;k<=t;k++)
  			  if (f[cur^1][k]) add(f[cur][k],(ll)f[cur^1][k]*(u-k));
  			s2++;
		  }
		else {
			int u=num[j].FR-s2,v=num[j].SE-i-n-s1,t=min(i,s1);
			for(int k=0;k<=t;k++)
			  if (f[cur^1][k]) {
			  	add(f[cur][k],(ll)f[cur^1][k]*(v+k));
			  	sub(f[cur][k+1],(ll)f[cur^1][k]*(u-k));
			  }
			s1++;
		}
	  }
	add(ans,f[cur][i]);
  }
  printf("%d\n",(ans+MOD)%MOD);
  return 0;
}