园艺问题
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
由题目可知有两种输入
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;
}