bzoj4361: isn （dp+树状数组+容斥）

Problem

给你一个序列，若此时这个序列不是非降的，那么从中删除一个数，知道删到非降为止。
问有多少种操作方案

Solution

定义 $f[i]$ 表示长度为 $i$ 的非降子序列个数
容斥一下， $ans=\sum f[i]\ast (n-i)!-f[i+1]\ast (n-i-1)!\ast (i+1)$
因为如果剩下的长度为 $i$ ，那就是把剩下的 $n-i$ 个数随机删去。但此时可能出现剩下 $i+1$ 个数的时候已经是非降的了，但是我们还是选了其中一个删去，那我们把这种情况删去就得到了上式。

下面问题转化成求 $f[i]$
设 $g[i][j]$ 表示长度为 $i$ 以 $j$ 结尾的个数，线段树/树状数组维护一下就好了。
$g[i][j]$ 可以省掉 $i$ 一维（只有前一次有关）

注意：ll…真的…数大的题贼讨厌…日常爆ll…

Code

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 2010
#define mod 1000000007
#define ll long long
int n,m,ans=0,a[N],b[N];
ll tree[N],f[N],g[N],fac[N];
inline void add(int x,int y){
	for(;x<=m;x+=x&(-x)) (tree[x]+=y)%mod;
}
inline ll query(int x){
	ll y=0;
	for(;x;x-=x&(-x)) (y+=tree[x])%mod;
	return y;
}
int main(){
	freopen("a.in","r",stdin);
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]),b[i]=a[i];
	sort(b+1,b+n+1);m=unique(b+1,b+n+1)-b-1;
	for(int i=1;i<=n;i++) a[i]=lower_bound(b+1,b+m+1,a[i])-b;
	for(int i=1;i<=n;i++) g[i]=1;
	f[1]=n;
	for(int i=2;i<=n;i++){
		ll res=0;
		memset(tree,0,sizeof(tree));
		for(int j=1;j<=n;j++){
			add(a[j],g[j]);
			if(j>=i) g[j]=(query(a[j])-g[j]+mod)%mod;
			else g[j]=0;
			res=(res+g[j])%mod;
		}
		f[i]=res;
	}
	fac[1]=1;
	for(int i=2;i<=n;i++) fac[i]=fac[i-1]*i%mod;
	for(int i=1;i<=n;i++){
		ans=(ans+(f[i]*fac[n-i]%mod-f[i+1]*fac[n-i-1]%mod*(i+1)%mod+mod)%mod)%mod; 
	}
	printf("%d\n",ans);
	return 0;
}