HDU4417（主席树计数）

解题思路：如果说用排序，那么复杂度就是O(mnlogn)会超时。用主席树的空间复杂度是O(nlogn)，时间复杂度是O(max(nlogn,mlogn).这道题稍微修改一下query函数即可。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#define N 100009
using namespace std;
struct node
{
	int l,r,num;
}s[N*20];
struct node2
{
	int id,v;
}a[N];
int rnk[N],rt[N],cnt,b[N];
bool cmp(node2 a,node2 b)
{
	return a.v<b.v;
}
void insert(int num,int &x,int l,int r)
{
	s[cnt++]=s[x];
	x=cnt-1;
	s[x].num++;
	if(l==r) return;
	int mid=(l+r)/2;
	if(num<=mid)
		insert(num,s[x].l,l,mid);
	else
		insert(num,s[x].r,mid+1,r);
}
int query(int u,int v,int w,int l,int r)
{
	int tp;
	if(r==w)
	{
		tp=s[v].num-s[u].num;
		//cout<<l<<" "<<r<<" "<<tp<<endl;
		return tp;
	}
	
	int k=s[s[v].l].num-s[s[u].l].num;
	int mid=(l+r)/2;
	if(mid<w)
	{
		tp=query(s[u].r,s[v].r,w,mid+1,r)+k;
		//cout<<l<<" "<<r<<" "<<tp<<endl;
		return tp;
	}else 
	{
		tp=query(s[u].l,s[v].l,w,l,mid);
		//cout<<l<<" "<<r<<" "<<tp<<endl;
		return tp;
	}
}
int main()
{
	int T,n,m,l,r,h;
	//freopen("t.txt","r",stdin);
	scanf("%d",&T);
	rt[0]=0;
	s[0].l=s[0].r=s[0].num=0;
	for(int ca=1;ca<=T;ca++)
	{
		printf("Case %d:\n",ca);
		scanf("%d%d",&n,&m);
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i].v);
			a[i].id=i;
		}
		sort(a+1,a+1+n,cmp);
		for(int i=1;i<=n;i++)
		{
			rnk[a[i].id]=i;
			b[i]=a[i].v;
		}
		cnt=1;
		for(int i=1;i<=n;i++)
		{
			rt[i]=rt[i-1];
			insert(rnk[i],rt[i],1,n);
		}
		while(m--)
		{
			scanf("%d%d%d",&l,&r,&h);
			int t=upper_bound(b+1,b+1+n,h)-(b+1);
			if(t==0)
			printf("0\n");
			else
			printf("%d\n",query(rt[l],rt[r+1],t,1,n));
		}
	}
	return 0;
}