JZOJ 5429 排列

排列

Description

有两个长度为$n$的排列$A$和$B$，定义排列的价值$f$($A$,$B$)为所有满足$A_i$>$B_i$的位置i的数量。
现给出$n$，$A$，$B$和$S$，其中$A$和$B$中有一些位置的数未知，问有多少种可能的填数的方案使得$f$($A$,$B$)=$S$。
若第$i$位上位置则为$0$。
答案可能很大，对$10^9$+$7$取模。

Data Constraint

$1$<=$S$<=$n$<=$4*10^3$
保证不存在一个位置$i$满足$A_i$=$0$且$B_i$=$0$

Solution

很显然可以把问题拆成两个完全一样的问题。
把$a$序列中缺的数取出来排好序，把$b$序列中失配的数取出来。
设$k_i$表示$a$序列中第$i$个数做多能够匹配到$b$序列中的第$k_i$个数。
设$f_{i,j}$表示$a$序列中的数做到了第$i$位，其中有$j$对匹配造成了贡献。
易得$f_{i,j}$=$f_{i-1,j}$+$f_{i-1,j-1}$*($k_i$-$j$+$1$)
设$G_i$表示整个$a$序列已处理完，保证有$j$的贡献的方案数(贡献至少为$j$)。
易得$G_i$=$f_{n,i}$*($n$-$i$)$!$
设$F_i$表示贡献恰好为$i$的方案数。
通过容斥原理可得$G_i$=$\sum_{j>=i}$$F_j$*$C_{j}^{i}$
移项可得$F_i$=$G_i$-$\sum_{j>i}$$F_j$*$C_j^i$
这样子问题就解决了，最后答案的计算只需合并两个子问题的答案即可。

Code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>

#define fo(i,j,l) for(int i=j;i<=l;i++)
#define fd(i,j,l) for(int i=j;i>=l;i--)

using namespace std;
typedef long long ll;
const ll N=45e2,mo=1e9+7;
ll f1[N],g1[N][N],f2[N],g2[N][N],a[N],b[N],c[N],d[N],e[N],least[N];
int n,m,j,k,l,i,o,p,u,v;
ll jc[N],ny[N],r1[N],r2[N];

ll ksm(ll o,ll t)
{
    ll yy=1;
    for(;t;t>>=1,o=o*o%mo)
    if(t%2)yy=yy*o%mo;
    return yy;
}

int min(int a,int b)
{if(a<b)return a;else return b;}

ll C(ll a,ll b)
{
    return (jc[a]*ny[b]%mo)*ny[a-b]%mo;
}

void work1()
{
    int k=0,v=0;
    fo(i,1,n)if(!r1[i])e[++k]=i;
    fo(i,1,k){
        for(;e[i]>d[v+1]&&v<k;)v++;
        least[i]=v;
    }
    fo(i,0,k)g1[i][0]=1;
    fo(i,1,k)
    fo(l,1,min(least[i],i))
    g1[i][l]=(g1[i-1][l]+g1[i-1][l-1]*(least[i]-l+1))%mo;
    f1[k]=g1[k][k];
    fd(i,k-1,1){
        f1[i]=g1[k][i]*jc[k-i]%mo;
        fo(j,i+1,k)f1[i]=(f1[i]-f1[j]*C(j,i))%mo;
    }
    f1[0]=jc[k];
    fo(i,1,k)f1[0]=(f1[0]-f1[i])%mo;
}

void work2()
{
    int k=0,v=0;
    fd(i,n,1)if(!r2[i])e[++k]=i;
    fo(i,1,k){
        for(;e[i]<c[v+1]&&v<k;)v++;
        least[i]=v;
    }
    fo(i,0,k)g2[i][0]=1;
    fo(i,1,k)
    fo(l,1,min(least[i],i))
    g2[i][l]=(g2[i-1][l]+g2[i-1][l-1]*(least[i]-l+1))%mo;
    f2[k]=g2[k][k];
    fd(i,k-1,1){
        f2[i]=g2[k][i]*jc[k-i]%mo;
        fo(j,i+1,k)f2[i]=(f2[i]-f2[j]*C(j,i))%mo;
    }
    f2[0]=jc[k];
    fo(i,1,k)f2[0]=(f2[0]-f2[i])%mo;
}

int main()
{
    cin>>n>>m;
    jc[0]=ny[0]=1;
    fo(i,1,n)jc[i]=jc[i-1]*i%mo,ny[i]=ksm(jc[i],mo-2);
    fo(i,1,n)scanf("%d",&a[i]);
    fo(i,1,n)scanf("%d",&b[i]);
    fo(i,1,n){
        if(!a[i])d[++u]=b[i]; r1[a[i]]++;
        if(!b[i])c[++v]=a[i]; r2[b[i]]++;
        if((a[i]*b[i]))m-=(a[i]>b[i]);
    }
    sort(d+1,d+u+1);    sort(c+1,c+v+1);
    fo(i,1,v/2)swap(c[i],c[v-i+1]);
    work1(); work2();
    ll ans;
    fo(i,1,m)
    ans=(ans+f1[i]*f2[m-i])%mo;
    ans=(ans+mo)%mo;
    printf("%lld",ans);
}