HDU 4325 Flowers(线段树+离散化)

Description
给出n朵花的绽放时刻与凋谢时刻,并给出m次查询,每次查询一个时刻有多少朵花同时开放
Input
第一行为一整数t表示用例组数,每组用例第一行为两个整数n和m分别表示花的数量和查询的次数,之后n行每行两个整数s和e分别表示该朵花的绽放时刻和凋谢时刻,最后m行每行一个整数x表示查询在x时刻有多少朵花同时开放(1<=t<=10,1<=n,m<=10^5,1<=s<=e<=10^9)
Output
对于每次查询,输出查询结果
Sample Input
2
1 1
5 10
4
2 3
1 4
4 8
1
4
6
Sample Output
Case #1:
0
Case #2:
1
2
1
Solution
将所有花的绽放时刻,凋谢时刻以及查询的时刻放在一起排序后离散化建线段树,那么每朵花的开放就是对线段树的区间覆盖,查询就是对线段树的单点更新,注意离散化的时候要通过增点来去重
Code

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
#define maxn 111111
struct Tree
{
    int left,right,sum,flag;
}T[4*maxn];
int t,n,m,s[maxn],e[maxn],cnt,x[4*maxn],q[maxn];
void build(int l,int r,int t)
{
    T[t].left=l;
    T[t].right=r;
    T[t].sum=T[t].flag=0;
    if(l==r) return ;
    int mid=(l+r)>>1;
    build(l,mid,2*t);
    build(mid+1,r,2*t+1);
}
void push_up(int t)
{
    T[t].sum=T[2*t].sum+T[2*t+1].sum;
}
void push_down(int t)
{
    if(T[t].flag)
    {
        T[2*t].flag+=T[t].flag;
        T[2*t+1].flag+=T[t].flag;
        T[2*t].sum+=(T[2*t].right-T[2*t].left+1)*T[t].flag;
        T[2*t+1].sum+=(T[2*t+1].right-T[2*t+1].left+1)*T[t].flag;
        T[t].flag=0;
    }
}
void update(int l,int r,int z,int t)
{
    if(T[t].left==l&&T[t].right==r)
    {
        T[t].flag+=z;
        T[t].sum+=(r-l+1)*z;
        return ;
    } 
    push_down(t);
    if(r<=T[2*t].right) update(l,r,z,2*t);
    else if(l>=T[2*t+1].left) update(l,r,z,2*t+1);
    else
    {
        update(l,T[2*t].right,z,2*t);
        update(T[2*t+1].left,r,z,2*t+1);
    }
    push_up(t);
}
int query(int tar,int t)
{
    if(T[t].left==tar&&T[t].right==tar) return T[t].sum;
    push_down(t);
    if(tar<=T[2*t].right) return query(tar,2*t);
    return query(tar,2*t+1);
}
int get(int tar)
{
    return lower_bound(x,x+cnt,tar)-x+1;
}
int main()
{
    int res=1;
    scanf("%d",&t);
    while(t--)
    {
        cnt=0;
        printf("Case #%d:\n",res++);
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)
        {
            scanf("%d%d",&s[i],&e[i]);
            x[cnt++]=s[i],x[cnt++]=e[i];
        }
        for(int i=1;i<=m;i++)
        {
            scanf("%d",&q[i]);
            x[cnt++]=q[i];
        }
        //离散化 
        sort(x,x+cnt);
        cnt=unique(x,x+cnt)-x;
        for(int i=cnt-1;i>0;i--)
            if(x[i]-x[i-1]>1) x[cnt++]=x[i-1]+1;
        sort(x,x+cnt);
        build(1,cnt,1);
        for(int i=1;i<=n;i++)
        {
            int l=get(s[i]),r=get(e[i]);//二分搜索到这朵花的绽放与凋谢时间 
            update(l,r,1,1);//覆盖这朵花的开放区间 
        }
        for(int i=1;i<=m;i++)
        {
            int tar=get(q[i]);//二分搜索到这个查询时刻是树中哪个节点 
            printf("%d\n",query(tar,1));
        }
    }
    return 0;
}

你可能感兴趣的:(HDU 4325 Flowers(线段树+离散化))