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\n",num1,num2);
}
}
}
return 0;
}