[BZOJ4361]isn（dp+树状数组+容斥原理）

题目描述

传送门

题解

首先dp
令f(i,j)表示选了i个，选到第j个，其中第j个必须选，选出来的顺次排列成一个非降子序列的方案数
这个是$O(n^2)$dp，$O(n)$递推，复杂度是$O(n^3)$的。可以用树状数组优化成$O(n^2logn)$
那么在整个子序列中选出来i个的方案数就是$\sum\limits_{k=1}^nf(i,k)$
假如说现在选了i个，也就是最终删除到这i个停止，那么剩下的n-i个数的顺序就无所谓了。但是这里面存在某些不合法的方案，就是删除到i+1个之后就已经满足非降就停止了，而这i+1中的哪一个作为最后一个删的都行。所以应该减去i+1的答案*(i+1)。

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
#define LL long long
#define Mod 1000000007
#define N 2005

int n,LSH,a[N],lsh[N];
LL ans,mul[N],C[N],f[N][N],ch[N];

void add(int loc,int val)
{
    for (int i=loc;i<=LSH;i+=i&-i)
        C[i]=(C[i]+val)%Mod;
}
LL query(int loc)
{
    LL ans=0;
    for (int i=loc;i>=1;i-=i&(-i))
        ans=(ans+C[i])%Mod;
    return ans;
}
int main()
{
    scanf("%d",&n);
    mul[0]=1;for (int i=1;i<=n;++i) mul[i]=mul[i-1]*(LL)i%Mod;
    for (int i=1;i<=n;++i) scanf("%d",&a[i]),lsh[++LSH]=a[i];
    sort(lsh+1,lsh+LSH+1);LSH=unique(lsh+1,lsh+LSH+1)-lsh-1;
    for (int i=1;i<=n;++i) a[i]=lower_bound(lsh+1,lsh+LSH+1,a[i])-lsh;
    for (int i=1;i<=n;++i)
    {
        memset(C,0,sizeof(C));
        if (i==1) add(1,1);
        for (int j=1;j<i;++j) add(a[j],f[i-1][j]);
        for (int j=i;j<=n;++j)
        {
            f[i][j]=query(a[j]);
            add(a[j],f[i-1][j]);
        }
    }
    for (int i=1;i<=n;++i)
        for (int j=1;j<=n;++j)
            ch[i]=(ch[i]+f[i][j])%Mod;
    for (int i=1;i<=n;++i)
    {
        ans+=(ch[i]*mul[n-i]%Mod-ch[i+1]*mul[n-i-1]%Mod*(i+1)%Mod)%Mod;
        ans=(ans%Mod+Mod)%Mod;
    }
    printf("%lld\n",ans);
}