BZOJ.4361.isn(DP 树状数组 容斥)

题目链接

长度为\(i\)的不降子序列个数是可以DP求的。

用\(f[i][j]\)表示长度为\(i\)，结尾元素为\(a_j\)的不降子序列个数。转移为\(f[i][j]=\sum f[i-1][k]\)，其中\(k\)满足\(k<j\)且\(a_k\leq a_j\)，可以用树状数组\(O(n^2\log n)\)解决。

那么长度为为\(i\)的不降子序列个数\(sum[i]=\sum_{j=i}^nf[i][j]\)。

比较麻烦的是得到不降序列后会立刻停止操作。如果没有这个限制，答案就是\(\sum_{i=1}^nsum[i]\times (n-i)!\)。

但是很简单的是，如果长为\(i\)的不降序列是由另一个不降序列继续删数得到的（即不合法的方案），那么这个方案数就是\(sum[i+1]\times (i+1)\times (n-i+1)!\)。

对每个\(i\)减掉不合法方案的贡献就可以了，即\(Ans=\sum_{i=1}^nsum[i]\times (n-i)!-sum[i+1]\times (i+1)\times (n-i+1)!\)。

//32268kb	1256ms
#include <cstdio>
#include <cctype>
#include <cstring>
#include <algorithm>
#define gc() getchar()
#define mod 1000000007
#define Mod(x) x>=mod&&(x-=mod)
typedef long long LL;
const int N=2005;
int fac[N],A[N],ref[N],f[N][N],g[N];
struct Bit
{
	int n,t[N];
	#define lb(x) (x&-x)
	inline void Add(int p,int v)
	{
		for(; p<=n; p+=lb(p)) t[p]+=v, Mod(t[p]);
	}
	inline int Query(int p)
	{
		LL res=0;
		for(; p; p^=lb(p)) res+=t[p];
		return res%mod;
	}
}T[N];
inline int read()
{
	int now=0;register char c=gc();
	for(;!isdigit(c);c=gc());
	for(;isdigit(c);now=now*10+c-'0',c=gc());
	return now;
}
inline int Find(int x,int r)
{
	int l=1,mid;
	while(l<r)
		if(ref[mid=l+r>>1]<x) l=mid+1;
		else r=mid;
	return l;
}
int main()
{
	int n=read(),mx=0;
	fac[0]=fac[1]=1;
	for(int i=2; i<=n; ++i) fac[i]=1ll*fac[i-1]*i%mod;
	for(int i=1; i<=n; ++i) ref[i]=A[i]=read();
	std::sort(ref+1,ref+1+n); int cnt=1;
	for(int i=2; i<=n; ++i) if(ref[i]!=ref[i-1]) ref[++cnt]=ref[i];
	for(int i=1; i<=n; ++i) mx=std::max(mx,A[i]=Find(A[i],cnt));
	for(int i=1; i<=n; ++i) f[1][i]=1;
	for(int i=2; i<=n; ++i)
	{
		T[i-1].n=mx, T[i-1].Add(A[i-1],f[i-1][i-1]);
		for(int j=i; j<=n; ++j)
			f[i][j]=T[i-1].Query(A[j]), T[i-1].Add(A[j],f[i-1][j]);
	}
	for(int i=1; i<=n; ++i)
	{
		LL sum=0;
		for(int j=i; j<=n; ++j) sum+=f[i][j];
		g[i]=1ll*sum%mod*fac[n-i]%mod;
	}
	LL ans=0;
	for(int i=1; i<=n; ++i) ans+=g[i]-1ll*g[i+1]*(i+1)%mod;
	printf("%lld\n",(ans%mod+mod)%mod);
	return 0;
}