首页 > 代码库 > HDU 4417 Super Mario 划分树/树状数组

HDU 4417 Super Mario 划分树/树状数组

题目大意:给定一个序列,求区间内小于等于某数的元素数量

首先区间排名一看就是划分树 不过和第k小不一样 我们需要做一些处理

第一种处理方式是二分答案 然后转换成区间第k小 好方法我喜欢 但是这里说的不是这种方法

首先建树,然后对于每个询问,我们讨论k与a[mid]的关系

若k<a[mid],则右子树一定没有小于等于k的数,我们进入左子树查找

若k>=a[mid],则左子树内一定所有数都小于等于k,于是我们将查询区间中进入左子树的元素的数量记入ans,然后查找右区间

递归退出条件是查询区间为空或节点为叶节点

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define M 100100
using namespace std;
int n,m,cnt,a[M],b[M],c[M];
int s[20][M];
void Build_Tree(int l,int r,int dpt)
{
	int i,mid=l+r>>1;
	int l1=l,l2=mid+1;
	int left=mid-l+1;
	if(l==r)
		return ;
	for(i=l;i<=r;i++)
		left-=(a[i]<c[mid]);
	for(i=l;i<=r;i++)
	{
		if(a[i]>c[mid]||a[i]==c[mid]&&!left)
			b[l2++]=a[i],s[dpt][i]=(i==l?0:s[dpt][i-1]  );
		else
			b[l1++]=a[i],s[dpt][i]=(i==l?1:s[dpt][i-1]+1),left-=(a[i]==c[mid]);
	}
	memcpy( a+l , b+l , sizeof(a[0])*(r-l+1) );
	Build_Tree(l,mid,dpt+1);
	Build_Tree(mid+1,r,dpt+1);
}
int Get_Ans(int l,int r,int dpt,int x,int y,int k)
{
	int mid=l+r>>1;
	int l1=(x==l?0:s[dpt][x-1]),l2=s[dpt][y];
	if(x>y)
		return 0;
	if(l==r)
		return a[mid]<=k;
	if(k<c[mid])
		return Get_Ans(l,mid,dpt+1,l+l1,l+l2-1,k);
	else
		return l2-l1+Get_Ans(mid+1,r,dpt+1,(mid+1)+(x-l-l1),(mid+1)+(y-l+1-l2)-1,k);
}
int main()
{
	int T,i,x,y,k;
	for(cin>>T;T;T--)
	{
		printf("Case %d:\n",++cnt);
		cin>>n>>m;
		for(i=1;i<=n;i++)
			scanf("%d",&a[i]),c[i]=a[i];
		sort(c+1,c+n+1);
		Build_Tree(1,n,0);
		for(i=1;i<=m;i++)
			scanf("%d%d%d",&x,&y,&k),printf("%d\n", Get_Ans(1,n,0,x+1,y+1,k) );
	}
}

后来看了看题解发现这题树状数组也能做。。。离线思想,先OTZ一下

我们把询问按照询问的值排序,对于每一个值,我们把小于等于这个值的元素位置都插进树状数组,然后查询(x-1,y]区间内有多少个元素即可

写完是140MS 划分树是109MS 没办法两次排序的常数有点大0.0

把反复memset改成时间标记的话是125MS 能快一些

然后别忘了输出Case 1...因为这点P事还贡献了一次WA

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define M 100100
using namespace std;
struct abcd{
	int x,y,num,pos;
	bool operator < (const abcd &x)const
	{
		return num < x.num;
	}
}q[M];
pair<int,int>a[M];
int n,m,cnt,ans[M];
int c[M],tim[M];
void update(int x)
{
	for(;x<=n;x+=x&-x)
	{
		if(tim[x]!=cnt)
			c[x]=0;
		tim[x]=cnt;
		c[x]++;
	}
}
int getans(int x)
{
	int re=0;
	for(;x;x-=x&-x)
		if(tim[x]==cnt)
			re+=c[x];
	return re;
}
int main()
{
	int T,i,j,x,y,k;
	for(cin>>T;T;T--)
	{
		printf("Case %d:\n",++cnt);
		cin>>n>>m;
		for(i=1;i<=n;i++)
			scanf("%d",&a[i].first),a[i].second=i;
		sort(a+1,a+n+1);
		for(i=1;i<=m;i++)
			scanf("%d%d%d",&q[i].x,&q[i].y,&q[i].num),q[i].pos=i;
		sort(q+1,q+m+1);
		j=1;
		for(i=1;i<=m;i++)
		{
			for(;j<=n&&a[j].first<=q[i].num;j++)
				update(a[j].second);
			ans[q[i].pos]=getans(q[i].y+1)-getans(q[i].x);
		}
		for(i=1;i<=m;i++)
			printf("%d\n",ans[i]);
	}
}


HDU 4417 Super Mario 划分树/树状数组