【SDUTOJ 3323】园艺有关问题 (离散化+线段树+离线数据处理)

【SDUTOJ 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输出一行，包含两个整数代表答案。

示例输入

示例输出

1 0
0 0

由题目可知有两种输入

1 d a b 表示第 [d,d+a-1]天开紫花 [d+a][d+a+b-1]开白花

2 x 询问第x天几紫几白

很容易想到线段树但10^9太大一共300000组区间也就是最多1200000个端点离散后叶子最多只需1200000即可这样把所有数据先存起来然后离线处理先把区间端点全存起来然后从小到大排序重标号建一个足够的线段树然后读到1就树上处理即可 2就树上查询注意由于没存x 所以查询的不一定是查到x叶子可能是包含x的一个最小区间的左端或右端但不影响

代码如下:

#include <bits/stdc++.h>

using namespace std;

typedef struct Node Node;
typedef struct Range Range;

struct Node//线段树
{
    int z,b;//节点紫花白花
};

struct Range//存数据
{
    int p,x,a,b,c,d;//p-1或2 1:[a,b] [c,d] 2:x
};

map <int,int> hs;//重标号哈希
int pt[1200012];//重标号
Node tr[4800048];//线段树
Range rg[300003];//预存数据离散处理
int tp;//重标号结点数

void SetTree(int site,int l,int r)//建空树
{
    if(l == r)
    {
        tr[site].z = tr[site].b = 0;
        return;
    }

    int mid = (l+r)>>1;
    SetTree(site<<1,l,mid);
    SetTree(site<<1|1,mid+1,r);
    tr[site].b = tr[site].z = 0;
}

void Add(int site,int l,int r,int ll,int rr,int z,int b)//更新树
{
    if(l == ll && r == rr)
    {
        tr[site].z += z;
        tr[site].b += b;
        //printf("%d %d z%d b%d\n",l,r,tr[site].z,tr[site].b);
        return;
    }
    int mid = (l+r)>>1;
    if(mid >= rr) Add(site<<1,l,mid,ll,rr,z,b);
    else if(mid < ll) Add(site<<1|1,mid+1,r,ll,rr,z,b);
    else
    {
        Add(site<<1,l,mid,ll,mid,z,b);
        Add(site<<1|1,mid+1,r,mid+1,rr,z,b);
    }
}

void Search(int site,int l,int r,int x)//查值
{
    if(l == r)
    {
        printf("%d %d\n",tr[site].z,tr[site].b);
        return;
    }
    int mid = (l+r)>>1;
    tr[site<<1].z += tr[site].z;
    tr[site<<1].b += tr[site].b;
    tr[site<<1|1].z += tr[site].z;
    tr[site<<1|1].b += tr[site].b;
    tr[site].z = tr[site].b = 0;

    if(pt[mid] >= x) Search(site<<1,l,mid,x);
    else Search(site<<1|1,mid+1,r,x);
}

int main()
{
    int n,a,b,x,i,j;

    while(~scanf("%d",&n))
    {
        hs.clear();
        tp = 0;
        for(i = 0; i < n; ++i)
        {
            scanf("%d",&rg[i].p);
            if(rg[i].p == 2)
            {
                scanf("%d",&rg[i].x);
            }
            else
            {
                scanf("%d %d %d",&rg[i].a,&a,&b);
                rg[i].b = rg[i].a+a-1;
                rg[i].c = rg[i].b+1;
                rg[i].d = rg[i].b+b;
                //[a,b] = [d,d+a-1] 紫花 [c,d] [d+a,d+a+b-1] 白花
                pt[tp++] = rg[i].a;
                pt[tp++] = rg[i].b;
                pt[tp++] = rg[i].c;
                pt[tp++] = rg[i].d;
            }
        }

        sort(pt,pt+tp);
        for(i = -1, j = 0; j < tp; ++j)//去重
        {
            if(i != -1 && pt[j] == pt[i]) continue;
            pt[++i] = pt[j];
            hs[pt[i]] = i;
        }
        tp = i;

        //for(i = 0; i <= tp; ++i) printf("%d %d\n",pt[i],hs[pt[i]]);

        SetTree(1,0,tp);
        for(i = 0; i < n; ++i)
        {
            if(rg[i].p == 1)
            {
                //printf("%d:%d %d:%d %d:%d %d:%d\n",rg[i].a,hs[rg[i].a],rg[i].b,hs[rg[i].b],rg[i].c,hs[rg[i].c],rg[i].d,hs[rg[i].d]);
                Add(1,0,tp,hs[rg[i].a],hs[rg[i].b],1,0);//[a,b] 紫花++
                Add(1,0,tp,hs[rg[i].c],hs[rg[i].d],0,1);//[c,d] 白花++
            }
            else
            {
                if(rg[i].x > pt[tp] || rg[i].x < pt[0]) puts("0 0");//由于x查到左右端点 可能查出范围 特判一下
                else Search(1,0,tp,rg[i].x);
            }
        }
    }

    return 0;
}

【SDUTOJ 3323】 园艺有关问题 (离散化+线段树+离线数据处理)