hdu_4325 Flowers（离散化+线段树）

http://acm.hdu.edu.cn/showproblem.php?pid=4325

题意：

给你n条线段，再给你m个点，问每个点分别被多少条线段覆盖。

思路：

先把线段和点一起离散化，再插入线段树，线段树功能：求和。插入线段时[l,r]区间每个值+1，查询时直接查第区间[i,i]的和，即i的值 。

我的代码：

#include<stdio.h>
#include<algorithm>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
int T,n,m,s,t,i,ans,js=1;
struct 
{
    int sum,add;
}ST[400001];
struct ls 
{
    int id,data,after; 
}a[200001];
void PushUp(int rt)
{
    ST[rt].sum=ST[rt<<1].sum+ST[rt<<1|1].sum;
}
void PushDown(int l,int r,int rt)
{
    ST[rt<<1].sum+=ST[rt].add*((l+r)/2-l+1);
    ST[rt<<1|1].sum+=ST[rt].add*(r-(l+r)/2);
    ST[rt<<1].add+=ST[rt].add;
    ST[rt<<1|1].add+=ST[rt].add;
    ST[rt].add=0;
}
void build(int l,int r,int rt)
{
    int m=(l+r)>>1;
    if(l==r){scanf("%I64d",&ST[rt].sum);ST[rt].add=0;return;}
    build(lson);
    build(rson);
    PushUp(rt);
}
void updata(int i,int j,int val,int l,int r,int rt)//[i,j)区间增加val 
{
    int m=(l+r)>>1;
    if(i<=l&&r<=j) {ST[rt].sum+=(r-l+1)*val;ST[rt].add+=val;return;}
    if(ST[rt].add) PushDown(l,r,rt);//开始忘记向下更新了，WA了好久。。。 
    if(i<=m) updata(i,j,val,lson);
    if(j>m) updata(i,j,val,rson);
    PushUp(rt);
}
void query(int L,int R,int l,int r,int rt)
{
    int m=(l+r)>>1;
    if(L<=l&&r<=R) {ans+=ST[rt].sum;return;}
    if(ST[rt].add) PushDown(l,r,rt);
    if(L<=m) query(L,R,lson);
    if(R>m) query(L,R,rson);
}
int cmp1(ls a,ls b)
{
    if(a.data==b.data) return a.id<b.id;
    return a.data<b.data;
}
int cmp2(ls a,ls b)
{
    return a.id<b.id;
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        for(i=0;i<2*n;i+=2)
        {
            scanf("%d%d",&a[i].data,&a[i+1].data);
            a[i].id=i;a[i+1].id=i+1;
        }
        for(i=2*n;i<m+2*n;i++)
        {
            scanf("%d",&a[i].data);
            a[i].id=i;
        }
        std::sort(a,a+2*n+m,cmp1);
        t=1;a[0].after=1;
        for(i=1;i<2*n+m;i++)
        {
            if(a[i].data==a[i-1].data) a[i].after=t;
            else a[i].after=++t;
        }
        std::sort(a,a+2*n+m,cmp2);
        for(i=0;i<=4*t;i++) ST[i].sum=ST[i].add=0;//初始化线段树 
        for(i=0;i<2*n;i+=2)
            updata(a[i].after,a[i+1].after,1,1,t,1);//值最大为t
        printf("Case #%d:\n",js++);
        for(i=2*n;i<m+2*n;i++)
        {            
            ans=0;
            query(a[i].after,a[i].after,1,t,1);
            printf("%d\n",ans);
        }
    }
    return 0;
}