[2016北京集训测试赛7]isn-[树状数组+dp+容斥]

Description

Solution

定义dp[i][j]为在1到i个数中选了j个数，并且保证选了i的选法总数。

dp[i][j]为所有满足A[k]>A[i]的k(k<i)的dp[k][j-1]之和。在处理完dp[i][j]后，在树状数组里A[i]位置填上dp[i][j-1]的值就好。这样可以优化一下复杂度。[A可能要离散化一下]

然后，容斥大法好~

定义g[x]为最终序列长度为x的方案数。由于x是从大变小，所有的g[i]都是已经处理完毕的了。

（似乎还有一种不用n²操作，直接扫一遍就好的方法，不知道是不是二项式反演）

Code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int mod=1e9+;
typedef long long ll;
int n,a[],t[],rk[];
bool cmp(int x,int y){return a[x]<a[y];}
ll dp[][],g[];

ll fac[],inv[];
void pre()
{
    fac[]=fac[]=inv[]=inv[]=;
    for (int i=;i<=n;i++)
    {
        fac[i]=fac[i-]*i%mod;
        inv[i]=(mod-mod/i)*inv[mod%i]%mod;
    }
    for (int i=;i<=n;i++) inv[i]=inv[i]*inv[i-]%mod;
}
ll C(int x,int y){return fac[y]*inv[x]%mod*inv[y-x]%mod;}

ll tree[];
void add(int id,ll x){for(;id<=n;id+=id&-id) tree[id]+=x,tree[id]%=mod;}
ll query(int id){ll re=;for(;id;id-=id&-id) re+=tree[id],re%=mod;return re;}
int main()
{
    scanf("%d",&n);
    pre();
    for (int i=;i<=n;i++)
    {
        scanf("%d",&a[i]);t[i]=i;
    }
    sort(t+,t+n+,cmp);
    int js=;
    a[]=-;
    for (int i=;i<=n;i++)
    {
        if (a[t[i]]!=a[t[i-]]) js++;
        rk[t[i]]=js;
    }
    for (int i=;i<=n;i++) dp[i][]=;
    for (int j=;j<=n;j++)
    {
        memset(tree,,sizeof(tree));
        add(rk[j-],dp[j-][j-]);
        for (int i=j;i<=n;i++)
        {
            dp[i][j]=query(rk[i]);
            add(rk[i],dp[i][j-]);
        }
    }

    g[n]=dp[n][n];
    for (int i=n-;i;i--)
    {
        for (int j=;j<=n;j++) g[i]+=dp[j][i]*fac[n-i]%mod,g[i]%=mod;
        for (int j=i+;j<=n;j++)
            g[i]-=C(i,j)*g[j]%mod*fac[j-i]%mod,g[i]%=mod;
    }
    ll ans=;
    for (int i=;i<=n;i++) ans=(ans+g[i]+mod)%mod;
    cout<<ans;
}