Jzoj5429 排列

有两个长度为n的排列A和B,定义排列的价值f(A,B)为所有满足A[i]>B[i]的位置i的数量。
现给出n,A,B和S,其中A和B中有一些位置的数未知,问有多少种可能的填数的方案使得f(A,B)=S

保证不存在一个位置i满足A[i]=0且B[i]=0

十分有趣的题目,是个神奇的组合数学+dp

我们先将问题拆成两个,因为题目保证没有A[i]=B[i]=0的位置

我们将所有A[i]=0的B[i]取出来,将这些元素排序得到x

再将所有不在A中的元素取出来排序得到y,设长度为m

那么,我们设一个dp

f[i][j]表示已经做到第i个数,其中已经保证有j个位置有贡献,其余位置情况不清楚的方案数

那么f[i][j]=f[i-1][j]+f[i-1][j-1]*(k-j){ k是满足x[k-1]<y[i]的最大的k }

我们令g[j]表示f[m][j]*(m-j)! 这样就得到了最终的方案数(因为对于其余位置不清楚的情况,可以随意调换位置)

我们现在可以用容斥原理来计算贡献恰好为j的g'[j],那么显然g'[m]=g[m]

让后可以得到g'[j]=g[j]-Σg'[i]*C(j,i) {j<i<=m} 倒序计算即可

让后类似的我们可以将B[i]=0的A[i]取出来做一次类似的dp,也求出一个g''

那么答案就是Σg'[i]*g''[i-s](0<=i<=S)

感觉讲的比较不清楚,可以好好理解一下

(十月份都是什么题啊!!!!!!!)

#pragma GCC optimize("O3")
#pragma G++ optimize("O3")
#include<stdio.h>
#include<string.h>
#include<algorithm>
#define N 4010 
#define LL long long 
#define M 1000000007 
using namespace std;
bool va[N],vb[N];
int a[N],b[N],n,m,s,x[N],y[N];
LL js[N],inv[N],f[N],g[N],Answer=0;
inline LL C(int n,int m){ return js[n]*inv[m]%M*inv[n-m]%M; }
inline LL pow(LL x,int k,LL S=1){
	for(;k;x=x*x%M,k>>=1) if(k&1) S=S*x%M;
	return S;
}
int main(){
	freopen("arrange.in","r",stdin);
	freopen("arrange.out","w",stdout);
	scanf("%d%d",&n,&s);
	for(int i=*js=*inv=1;i<=n;++i) js[i]=js[i-1]*i%M;
	for(inv[n]=pow(js[n],M-2);;) break;
	for(int i=n;i;--i) inv[i-1]=inv[i]*i%M; 
	for(int i=1;i<=n;++i){
		scanf("%d",a+i);
		va[a[i]]=1;
	}
	for(int i=1;i<=n;++i){
		scanf("%d",b+i); 
		vb[b[i]]=1;
		if(a[i]&&b[i]&&a[i]>b[i]) --s;
	}
	m=0;
	for(int i=1;i<=n;++i) if(!a[i]) x[++m]=b[i];
	sort(x+1,x+1+m); m=0;
	for(int i=1;i<=n;++i) if(!va[i]) y[++m]=i;
	f[0]=1;
	for(int i=1,k=1;i<=m;++i){
		for(;k<=m && y[i]>x[k];++k);
		for(int j=i;j;--j) f[j]=(f[j]+f[j-1]*(k-j))%M;
	}
	for(int i=m;~i;--i){
		f[i]=(f[i]*js[m-i])%M;
		for(int j=i+1;j<=m;++j) f[i]=(f[i]+M-f[j]*C(j,i)%M)%M;
	}
	m=0;
	for(int i=1;i<=n;++i) if(!b[i]) x[++m]=a[i];
	sort(x+1,x+1+m); m=0; 
	for(int i=1;i<=n;++i) if(!vb[i]) y[++m]=i;
	for(int i=1;i<=m;++i) swap(x[i],y[i]);
	g[0]=1;
	for(int i=1,k=1;i<=m;++i){
		for(;k<=m && y[i]>x[k];++k);
		for(int j=i;j;--j) g[j]=(g[j]+g[j-1]*(k-j))%M;
	}
	for(int i=m;~i;--i){
		g[i]=(g[i]*js[m-i])%M;
		for(int j=i+1;j<=m;++j) g[i]=(g[i]+M-g[j]*C(j,i)%M)%M;
	}
	for(int i=0;i<=s;++i) Answer=(Answer+f[i]*g[s-i]%M)%M;
	printf("%lld\n",Answer);
}

posted @ 2017-11-08 20:30  扩展的灰(Extended_Ash)  阅读(122)  评论(0编辑  收藏  举报