Gym 102091K The Stream of Corning 2【线段树】

<题目链接>

题目大意：

进行两种操作：1.给定一个数的出现时间、价值、消失时间；

　　　　　　　2.进行一次询问，问你当前时间，第K大的数的价值。

解题分析：

采用离线集中处理，将每个数的出现时间和它的消失时间分组存下，并且存下所有的询问。然后就是依次进行所有询问，将这些询问时间点之前的点插入，将已经消失的点删除。因为算了一下复杂度感觉能过，所以就没有用离散化，线段树的代表区间大小就是1~1e6，线段树的节点维护的信息就是这个节点所对应区间的有效节点个数。

#include <bits/stdc++.h>
using namespace std;
 
const int N = 1e5+ , MAX = 1e6+;
 
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
struct Node{
    int time,val;
    Node (int _time=,int _val=):time(_time),val(_val){}
    bool operator < (const Node & tmp) const { return time<tmp.time; }
}node1[N],node2[N],d[N];
 
int tr[MAX<<];
 
template<typename T>
inline void read(T&x){
    x=;int f=;char ch=getchar();
    while(ch<'' ||ch>''){ if(ch=='-')f=-; ch=getchar(); }
    while(ch>='' && ch<=''){ x=x*+ch-''; ch=getchar(); }
    x*=f;
}
 
inline void Pushup(int rt){ tr[rt]=tr[rt<<]+tr[rt<<|]; }
void update(int rt,int l,int r,int loc,int val){
    if(l==r){ tr[rt]+=val;return; }
    int mid=l+r>>;
    if(loc<=mid)update(lson,loc,val);
    if(loc>mid)update(rson,loc,val);
    Pushup(rt);
}
int query(int rt,int l,int r,int k){
    if(l==r)return l;
    int mid=l+r>>;
    if(tr[rt<<]>=k)return query(lson,k);
    else if(tr[rt<<]<k)return query(rson,k-tr[rt<<]);    //左子树不够k个，则第k个一定在右子树
}
int main(){
    int T,ncase=;scanf("%d",&T);
    while(T--){
        printf("Case %d:\n",++ncase);
        int q;scanf("%d",&q);
        int cnt1=,cnt2=,cnt3=;
        while(q--){
            int op;read(op);
            if(op==){
                int s,e,w;read(s);read(w);read(e);
                cnt1++,cnt2++;
                node1[cnt1]=Node(s,w);     //node1记录开始时间
                node2[cnt2]=Node(e,w);     //node2记录结束时间
            }else {
                int now,k;read(now),read(k);
                cnt3++;
                d[cnt3].time=now,d[cnt3].val=k;     //记录下询问数组
            }
        }
        sort(node1+,node1++cnt1);sort(node2+,node2++cnt2); sort(d+,d++cnt3);
        int p1=,p2=;    //遍历‘建立’和‘删除’这两个数组的指针
        memset(tr,,sizeof(tr));     //建树
        for(int i=;i<=cnt3;i++){
            int now=d[i].time,k=d[i].val;
            while(p1<=cnt1 && node1[p1].time<=now){   //将当前时间前存活的的点插入线段树
                update(,,MAX,node1[p1].val,);
                p1++;
            }
            while(p2<=cnt2 && node2[p2].time<now){    //将当前时间前死亡的点删除线段树
                update(,,MAX,node2[p2].val,-);
                p2++;
            }
            if(tr[]<k){ puts("-1");continue; }    //如果不足k个存活，直接输出-1
            printf("%d\n",query(,,MAX,k));
        }
    }
}