LightOJ - 1089 线段树+离散化

题意：

给定n段区间，q个查询，询问每个点在上述区间中出现了多少次。

分析：

普通离散化，直接将每个区间端点还有查询点加入进行离散化就好。但是注意这题线段树不知道为什么要开这么大，可能一组数据点集过于稠密？坑。

#include<bits/stdc++.h>

using namespace std;

const int maxn=1e5+10;

int id[maxn*3],que[maxn];

struct node{
    int l,r;
}p[maxn];

struct Node{
    int val;
    int mark;
}tree[maxn<<4];

void update(int l,int r,int i){
    if(!tree[i].mark) return;
    tree[i<<1].val+=tree[i].mark;
    tree[i<<1|1].val+=tree[i].mark;
    tree[i<<1].mark+=tree[i].mark;
    tree[i<<1|1].mark+=tree[i].mark;
    tree[i].mark=0;
}

void change(int l,int r,int i,int x,int y){
    if(r<x||l>y) return;
    if(x<=l&&r<=y) {
        tree[i].val++;
        tree[i].mark++;
        return;
    }
    update(l,r,i);
    int mid=l+r>>1;
    change(l,mid,i<<1,x,y);
    change(mid+1,r,i<<1|1,x,y);
}

int query(int l,int r,int i,int x){
    update(l,r,i);
    if(l==r){
        return tree[i].val;
    }
    int mid=l+r>>1;
    if(x<=mid){
        return query(l,mid,i<<1,x);
    }else return query(mid+1,r,i<<1|1,x);
}

int main(){
    int T;
    scanf("%d",&T);
    for(int cs=1;cs<=T;cs++){
        printf("Case %d:\n",cs);
        int n,q;
        scanf("%d%d",&n,&q);
        int l,r,x;
        int cnt=0;
        for(int i=0;i<n;i++){
            scanf("%d%d",&l,&r);
            id[++cnt]=l;id[++cnt]=r;
            p[i].l=l;p[i].r=r;
        }
        for(int i=0;i<q;i++){
            scanf("%d",&que[i]);
            id[++cnt]=que[i];
        }
        sort(id+1,id+1+cnt);
        cnt=unique(id+1,id+1+cnt)-(id+1);
        memset(tree,0,sizeof tree);
        for(int i=0;i<n;i++){
            int xx=lower_bound(id+1,id+1+cnt,p[i].l)-id;
            int yy=lower_bound(id+1,id+1+cnt,p[i].r)-id;
            change(1,cnt,1,xx,yy);
        }
        for(int i=0;i<q;i++){
            int xx=lower_bound(id+1,id+1+cnt,que[i])-id;
            printf("%d\n",query(1,cnt,1,xx));
        }
    }
    return 0;
}