权值线段树找第k大

2019暑期集训第二次校赛-B.简易版第k大
链接：https://ac.nowcoder.com/acm/contest/1076/B
来源：牛客网

题目描述
有一个长度为n的序列，序列每个元素取值范围[1, 1e6]，现在有q次操作，每次操作：
1 k：查询整个序列第 k 大元素（指的是从小到大排序后第 k 个元素）
2 x y：将下标为x（下标从1开始）的元素值修改为y
输入描述:
第一行输入两个数n，q (1 <= n, q <=3e5)

第二行输入n个数，表示序列的元素ai（1 <= ai <= 1e6）

接下来q行每行输入 1 k 或者 2 x y （1 <= x <= n，1 <= y <= 1e6）
输出描述:
对于每次操作 1 k，输出一个数表示答案
示例1
输入
复制
5 3
1 2 3 4 5
1 3
2 1 5
1 3
输出
复制
3
4

思路
权值线段树，利用底层节点区间编号L表示a[i]的值，节点的权值表示对应值的个数。父亲节点权值为两个儿子节点权值之和，建树，当左儿子权值大于等于k时第k大在做儿子管辖区间，往左查；当左儿子权值小于k时，第k大在右儿子管辖的区间，k-tr[id<<1],往右查。
题解

#include<bits/stdc++.h>
using namespace std;
const int MAXN=1e6;
int a[300010],tr[4*MAXN + 5];
void updata(int L,int R,int id,int res,int flag)
{
    if(L==R)
    {
        tr[id]+=flag;
        return ;
    }
    int m=(L+R)>>1;
    if(res<=m)updata(L,m,id<<1,res,flag);
    if(res>m)updata(m+1,R,(id<<1)+1,res,flag);
    tr[id]=tr[id<<1]+tr[(id<<1)+1];
    
}
int query(int L,int R,int id,int k)
{
	if(k<=0)return 0;
    if(L==R)
    {
        return L;
    }
    int m=(L+R)>>1;
    if(tr[id<<1]>=k) return query(L,m,id<<1,k);
    else return query(m+1,R,(id<<1)+1,k-tr[id<<1]);
    
}



int main()
{
    memset(tr,0,sizeof(tr));
    int n,q;
    cin>>n>>q;
    int mx=0;
    for(int i=1;i<=n;i++) {
        cin>>a[i];
        updata(1, MAXN * 4,1,a[i],1);     
    }
    
    int opt,k,x,y;
    
	for(int i=1;i<=q;i++)
	{
		cin>>opt;
		if(opt==1)
		{
		    cin>>k;
			cout<<query(1,MAXN * 4,1,k)<<endl;	
		}
		if(opt==2)
		{
			cin>>x>>y;		
			updata(1,MAXN * 4,1,a[x],-1);
			a[x]=y;
			updata(1,MAXN * 4,1,y,1);
		}
		
		
	  }  

    
}