【ybtoj 高效进阶 4.2】【树状数组】 严格上升子序列

【ybtoj 高效进阶 4.2】【树状数组】 严格上升子序列

题目

---

解题思路

由题得，可用DP解
f[i][j]表示以第i个数为结尾长度为j的上升子序列
当长度为1时，方案数为1
f[i][j]=1;
当长度>1时，方案数为以小于a[i]的所有数结尾长为j-1的和
f[i][j]+=f[k][j-1];

时间复杂度挺让人绝望的😫
将数组离散化
建m个树状数组（二维表示），维护长度1~m的放案数
用来优化DP

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int mo=1000000007;
struct lzf{
	int x,i;
}a[1010];
int n,m,w,t;
long long c[1010][1010],f[1010][1010],ans=0; 
bool cmp(lzf f,lzf y)
{
	 if (f.x==y.x)
	    return f.i>y.i;
		else return f.x<y.x;
}
void add(int x,int y,int w)
{
	 for (;x<=n;x+=x&(-x))
	     c[x][y]=(c[x][y]+w)%mo;
}
int sum(int x,int y)
{
	long long da=0;
	for (;x;x-=x&(-x))
	    da=(da+c[x][y])%mo;
    return da%mo;
}
int main()
{
	scanf("%d",&w);
	while (w--)
	{
		  t++,ans=0;
		  memset(c,0,sizeof(c));
		  scanf("%d%d",&n,&m);
		  for (int i=1;i<=n;i++)
		  {
		  	  scanf("%d",&a[i].x);
		  	  a[i].i=i;
		  }
		  sort(a+1,a+n+1,cmp);
		  for (int i=1;i<=n;i++)
		      for (int j=1;j<=m;j++)
		      {
		      	  if (j==1) f[i][j]=1;
		      	     else f[i][j]=sum(a[i].i-1,j-1);  //求出以比a[i]小的数结尾，长度为j-1的和
		      	  add(a[i].i,j,f[i][j]);  //将答案放入树状数组
			  }
		   for (int i=m;i<=n;i++)
		       ans=(ans+f[i][m])%mo;
		   printf("Case #%d: %lld\n",t,ans);
	}	
}