SGU Ice-cream Tycoon 线段树+离散化

已单价划分区间，线段树节点保存数量和总价。优先处理单价较小的（即左节点）。

成段更新，延迟标记。这个题目存在卖与不卖两种情况，所以第一遍先查询一下到底能不能卖，如果能卖，再进行处理（下面代码的deal函数）。

因为数据范围很大，需要用__int64 保存，但是内存给的很小，所以我们需要离散化处理。把单价离散化，在具体计算的时候再搞回来。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define MAXN 400009
#define nmaxn 100009
#define LL __int64
struct node
{
bool col;
LL num;
LL tot;
} sum[MAXN];
LL se[nmaxn];
int k;
//离散化，二分查找
int sea(int l,int r,LL va)
{
while(l<=r)
{
int mid=(l+r)>>1;
if(se[mid]==va)return mid;
else if(se[mid]<va)l=mid+1;
else r=mid-1;
}
}
//向上更新
void update(int rt)
{
sum[rt].num=sum[rt<<1].num+sum[rt<<1|1].num;
sum[rt].tot=sum[rt<<1].tot+sum[rt<<1|1].tot;
}
//向下更新延迟标记
void downdate(int rt)
{
if(sum[rt].col==1)
{
sum[rt<<1].col=sum[rt<<1|1].col=1;
sum[rt].col=0;
sum[rt<<1].num=sum[rt<<1|1].num=0;
sum[rt<<1].tot=sum[rt<<1|1].tot=0;
}
return ;
}
//查询卖不卖
bool query(int l,int r,int rt,LL number,LL totsum)
{
if(totsum<0)return false;
if(sum[rt].num<number)return false;
else if(sum[rt].num==number&&sum[rt].tot>totsum)return false;
else if(sum[rt].num==number&&sum[rt].tot<=totsum)return true;
else
{
if(l==r)
{
if((LL)se[l]*number>totsum)return false;
else return true;
}
downdate(rt);
int m=(l+r)>>1;
if(sum[rt<<1].num>=number)return query(lson,number,totsum);
else return query(rson,number-sum[rt<<1].num,totsum-sum[rt<<1].tot);
}
}
//如果能卖，把相应的东西卖掉，加入在【l，r】的单价范围内都能卖，那么在这段线段树上进行延时标记
void deal(int l,int r,int rt,LL number,LL totsum)
{
if(number==0||totsum<=0)return ;
if(l==r)
{
sum[rt].num-=number;
sum[rt].tot-=(LL)se[l]*number;
return;
}
int m=(l+r)>>1;
downdate(rt);
if(sum[rt<<1].num>number)deal(lson,number,totsum);
else
{
sum[rt<<1].col=1;
deal(rson,number-sum[rt<<1].num,totsum-sum[rt<<1].tot);
sum[rt<<1].num=sum[rt<<1].tot=0;
}
update(rt);
return ;
}
//进货的情况 在单价为p的叶子节点上更新信息
void add(int l,int r,int rt,int p,LL number)
{
if(l==r)
{
sum[rt].num+=number;
sum[rt].tot+=(LL)se[p]*number;
return ;
}
downdate(rt);
int m=(l+r)>>1;
if(m>=p)add(lson,p,number);
else add(rson,p,number);
update(rt);
return ;
}
//建树，初始化线段树
void build(int l,int r,int rt)
{
if(l==r)
{
sum[rt].col=sum[rt].num=sum[rt].tot=0;
return ;
}
int m=(l+r)>>1;
build(lson);
build(rson);
update(rt);
}
LL y[nmaxn],x[nmaxn];
char s1[nmaxn][30];
int main()
{
LL n,m,i,j,z;
int num=1;
build(1,nmaxn,1);
i=0;
while(scanf("%s%I64d%I64d",s1[i],&x[i],&y[i])!=-1)
{
//离散化单价
if(strcmp(s1[i],"ARRIVE")==0)
{
se[num++]=y[i];
}
i++;
}
z=i;
//离散化
sort(se+1,se+num);
k=1;
for(i=2; i<num; i++)
{
if(se[i]!=se[i-1])se[++k]=se[i];
}
//依次输出结果
for(i=0; i<z; i++)
{
if(strcmp(s1[i],"ARRIVE")==0)
{
int temy=sea(1,k,y[i]);
add(1,nmaxn,1,temy,x[i]);
}
else
{
if(query(1,nmaxn,1,x[i],y[i]))
{
printf("HAPPY\n");
deal(1,nmaxn,1,x[i],y[i]);
}
else printf("UNHAPPY\n");
}
}
return 0;
}
感谢ink同学的无私帮助