SUDTOJ 3323园艺问题 (线段树)

园艺问题
Time Limit: 1000ms Memory limit: 65536K 有疑问？点这里^_^
题目描述

本巨养了一盆双色茉莉。这种花有一种特点：第i朵花在第Di天盛开，刚开时是紫色的，Ai天之后会变成白色，再过Bi天就会凋谢。如Di = 3，Ai = 4，Bi = 5，那么在第3至6天为紫色，第7至11天为白色，第11天之后就凋谢了。

现在给出一些事件，你需要按要求给出答案。

事件1：在第Di天开了一朵花，这朵花Ai天后变成白色，再过Bi天就会凋谢。

事件2：询问在第X天时，紫色花朵和白色花朵各有多少。
输入

输入包含多组。对于每组数据：

第一行包含一个整数n (1 <= n <= 300,000)。

接下来的n行，为下述两种格式的一种，分别代表事件1和事件2。

1 Di Ai Bi

2 X

对于所有数据有：1 <= Di,Ai,Bi <= 1,000,000,000 ,1 <= Ai <= 3,000,000,000；
输出

对于每个事件2输出一行，包含两个整数代表答案。
示例输入

4
1 3 2 3
2 4
1 2 1 1
2 10

示例输出

1 0
0 0
周赛的时候没有想到可以先全部输入进行离散化,再进行建树,我啸的一句话惊醒了我

#include <bits/stdc++.h>
#define LL long long
#define esp 1e-8
#define INF 0x3f3f3f3f
#define Fread() freopen("../in.in","r",stdin)
#define Fwrite() freopen("../out.out","w",stdout)

using namespace std;

const int Max = 301000;

typedef struct Flow
{
    LL l,r,L,R;
}Flow;

union Flower
{
    Flow F;
    LL Q;
};
typedef struct Seg
{
    int sum1;
    int sum2;
} Tree;
int Sta[Max];
Flower FF[Max];
Tree T[Max*5];
LL Da[Max*2];
int top2,top3;
LL a,b,d;
int c;
int num1,num2;
void Update_down(int site,int L,int R)//向下更新
{
    if(L==R)
    {
        return ;
    }
    T[site<<1].sum1+=T[site].sum1;
    T[site<<1].sum2+=T[site].sum2;
    T[site<<1|1].sum1+=T[site].sum1;
    T[site<<1|1].sum2+=T[site].sum2;
    T[site].sum1=T[site].sum2=0;
}
void Build(int L,int R,int site)//建树
{
    T[site].sum1=0;
    T[site].sum2=0;
    if(L==R)
    {
        return ;
    }
    int mid=(L+R)>>1;
    Build(L,mid,site<<1);
    Build(mid+1,R,site<<1|1);
}
void Update(int L,int R,int l,int r,int site,int ans)//更新
{

    if(L==l&&R==r)
    {
        if(ans==1)
        {
            T[site].sum1++;
        }
        else
        {
            T[site].sum2++;
        }
        return ;
    }
    int mid=(L+R)>>1;
    if(r<=mid)
    {
        Update(L,mid,l,r,site<<1,ans);
    }
    else if(l>mid)
    {
        Update(mid+1,R,l,r,site<<1|1,ans);
    }
    else
    {
        Update(L,mid,l,mid,site<<1,ans);
        Update(mid+1,R,mid+1,r,site<<1|1,ans);
    }
}
void Query(int L,int R,int x,int site)//查询
{
    if(L==R)
    {
        num1=T[site].sum1;
        num2=T[site].sum2;
        return ;
    }
    Update_down(site,L,R);
    int mid=(L+R)>>1;
    if(x<=mid)
    {
        Query(L,mid,x,site<<1);
    }
    else
    {
        Query(mid+1,R,x,site<<1|1);
    }

}
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        vector<LL>Arr;
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&Sta[i]);
            if(Sta[i]==1)
            {
                scanf("%lld %lld %lld",&d,&a,&b);
                FF[i].F.l=d;
                FF[i].F.r=d+a-1;
                FF[i].F.L=d+a;
                FF[i].F.R=d+a+b-1;
                Arr.push_back(d);
                Arr.push_back(d+a-1);
                Arr.push_back(d+a);
                Arr.push_back(d+a+b-1);
            }
            else
            {
                scanf("%lld",&FF[i].Q);
                Arr.push_back(FF[i].Q);
            }
        }
        sort(Arr.begin(),Arr.end());
        Arr.erase(unique(Arr.begin(),Arr.end()),Arr.end());//去重
        int ans=0;
        map<LL,int>Ma;
        for(int i=0; i<Arr.size(); i++)
        {
            Ma[Arr[i]]=++ans;//编号
        }
        Build(1,ans,1);
        for(int i=1;i<=n;i++)
        {
            if(Sta[i]==1)
            {
                Update(1,ans,Ma[FF[i].F.l],Ma[FF[i].F.r],1,1);
                Update(1,ans,Ma[FF[i].F.L],Ma[FF[i].F.R],1,2);
            }
            else
            {
                num1=num2=0;
                Query(1,ans,Ma[FF[i].Q],1);
                printf("%d %d
",num1,num2);
            }
        }
    }
    return 0;
}