nyoj 600——花儿朵朵——【离散化、线段树插线问点】

花儿朵朵

时间限制：1000 ms  |  内存限制：65535 KB

难度：5

描述

春天到了，花儿朵朵盛开，hrdv是一座大花园的主人，在他的花园里种着许多种鲜花，每当这个时候，就会有一大群游客来他的花园欣赏漂亮的花朵，游客们总是会询问，某个时间有多少种花儿同时在盛开着？hrdv虽然知道每种花儿的开花时间段，但是他不能很快的答出游客的问题，你能编写一个程序帮助他吗？

输入

第一行有个整数t,表示有t组测试数据，每组测试数据第一行为两个整数n,m(0<n<100000，0<m<100000);随后有n行，每一行有两个整数x,y(0<x<y<1000000000)，表示这一种花的盛开时间是从x到y；随后有m行，每行有一个整数，代表游客询问的时间。

输出

对于每次游客的询问，输出一个整数在单独的一行，表示这个时间盛开的花有多少种。

样例输入

2
1 1
5 10
4
2 3
1 4
4 8
1
4
6

样例输出

0
1
2
1

解题思路：先把所给的值都离散化后再操作。然后就是插线问点了。

#include<bits/stdc++.h>
using namespace std;
const int maxn=120000;
#define mid (L+R)/2
#define lson rt*2,L,mid
#define rson rt*2+1,mid+1,R
int a[maxn*2],aa[maxn*2],a_[maxn*2];
int x[maxn*2],y[maxn*2],q[maxn];
int f_num[maxn*4];
int discretization(int *tm,int l,int r,int key){
    int m;      //离散化
    while(l<=r){
        m=(l+r)/2;
        if(tm[m]<key){
            l=m+1;
        }else if(tm[m]>key){
            r=m-1;
        }else{
            return m;
        }
    }
}
void update(int rt,int L,int R,int l_ran,int r_ran){
    if(l_ran<=L&&R<=r_ran){ //区间更新
        f_num[rt]+=1;
        return ;
    }
    if(l_ran<=mid){
        update(lson,l_ran,r_ran);
    }
    if(r_ran>mid){
        update(rson,l_ran,r_ran);
    }
}
void PushDown(int rt){
    f_num[rt*2]+=f_num[rt];
    f_num[rt*2+1]+=f_num[rt];
    f_num[rt]=0;
}
int query(int rt,int L,int R,int v){    //单点查询
    if(L==R){
        return f_num[rt];
    }
    if(f_num[rt]!=0)
    PushDown(rt);
    if(mid>=v){
        return query(lson,v);
    }
    if(mid<v){
        return query(rson,v);
    }
}
int main(){
    int t;
    scanf("%d",&t);
    while(t--){
        memset(f_num,0,sizeof(f_num));
        int n,m,tmp;
        scanf("%d%d",&n,&m);
        int num_a=n*2+m;
        for(int i=0;i<num_a;i++){
            scanf("%d",&tmp);
            a[i]=tmp;
            aa[i]=tmp;
        }
        sort(a,a+num_a);
        int num=0;
        a_[num++]=a[0];
        for(int i=1;i<num_a;i++){
            if(a[i]!=a[i-1]){
                a_[num++]=a[i];
            }
        }
        int num_x=0,num_y=0,num_q=0,nn=0;
        for(int i=0;i<n*2;i++){
            if(i%2==0){
                x[num_x++]=discretization(a_,0,num-1,aa[i])+1;
                if(nn<x[num_x-1]){
                    nn=x[num_x-1];
                }
            }else{
                y[num_y++]=discretization(a_,0,num-1,aa[i])+1;
                if(nn<y[num_y-1]){
                    nn=y[num_y-1];
                }
            }
        }
        for(int i=n*2;i<num_a;i++){
            q[num_q++]=discretization(a_,0,num-1,aa[i])+1;
        }
        for(int i=0;i<n;i++){
            update(1,1,nn,x[i],y[i]);
        }
        for(int i=0;i<num_q;i++){
            int ans=query(1,1,nn,q[i]);
            printf("%d\n",ans);
        }
    }
    return 0;
}