HDU 4417 Super Mario--主席树+二分（查询前K大和）

STL二分

还不是很能理解不同版本相减的操作，因此用个pushup证明自己是蒟蒻

这样add只用更新一个点，再pushup回溯，每个结点就是区间和，而且query好写一些

注意这题可以离线，因此可以二分找出要求的是第几大，但要特判找不到的情况（输出0）

#include<cstdio>
#include<algorithm>
#include<iostream> 
using namespace std;
#define lson l,m,ls[rt]
#define rson m+1,r,rs[rt] 
typedef long long ll;
const int maxn = 1e5 + 5;
int ls[maxn*30],rs[maxn*30],tot,rts[maxn];//tot为总结点数 ，rts为每个位置的线段树的根
int t[maxn*30];//重点！权值线段树，存的是每个数出现次数
int a[maxn],num[maxn];//原数组和离散化的

void build(int l,int r,int &rt)
{
	rt = ++tot;
	t[rt] = 0; 
	if(l==r)return ;	
		int m = l+r>>1;
	build(lson); 
    build(rson);
		

	
}

void add(int p,int C,int l,int r,int &rt,int lst)//p为数的大小，因为已经离散化排序了 
{
	rt = ++tot;
	ls[rt] = ls[lst],rs[rt] = rs[lst];
//	t[rt] = t[lst] + C;
	if(l==r){
		t[rt] = t[lst] + C;	
		//上一个版本的结点拷贝信息，一般C=1，表示多了一个数，也就出现次数+1
		return ;
	}	
		int m = l+r>>1;
	if(p<=m)
	   add(p,C,lson,ls[lst]);//加个ls[rt]= 
	else
	   add(p,C,rson,rs[lst]);
	   
	   t[rt]=t[ls[rt]]+t[rs[rt]];//pushup

}
 
 int query(int l,int r,int rt,int k)//单点查询 ，好理解 
 {
 	if(l==r)return t[rt];
 	int m=l+r>>1;
 	int ret=0;
 	if(k<=m)//x<=m还是k<=m 
 	  ret+=query(lson,k);
 	  else{
 	  	       ret+=t[ls[rt]];//右边有一部分，记得加上左边 
 	  		   ret+=query(rson,k);
	   }
 
 	   
 	   return ret;
 }


int N,M,c,x,y,k;
int main()
{
	int T;
	cin>>T;
	
	for(int i=1;i<=T;i++)
	{
		scanf("%d%d",&N,&M);
		tot = 0;	
		for(int i=1;i<=N;++i)
		{
			scanf("%d",&a[i]);
			num[i] = a[i];
		}
		
		sort(num+1,num+N+1);
		int cnt = unique(num+1,num+N+1)-(num+1);
		build(1,cnt,rts[0]);
		for(int i=1;i<=N;++i)
			a[i] = lower_bound(num+1,num+cnt+1,a[i])-num;
		
		for(int i=1;i<=N;++i)
			add(a[i],1,1,cnt,rts[i],rts[i-1]);
			
		printf("Case %d:\n",i);		
		while(M--)
		{
			int o;
			scanf("%d%d%d",&x,&y,&o);
			int k=upper_bound(num+1,num+cnt+1,o)-num-1;//cout<<k<<"!"<<endl; 
			if(k>0)printf("%d\n",query(1,cnt,rts[y+1],k)-query(1,cnt,rts[x],k));
                   else  printf("0\n");
		}
	}	
	return 0;
}