SGU 311 Ice-cream Tycoon(线段树)

题目链接：http://acm.sgu.ru/problem.php?contest=0&problem=311

题意：一个商店，有两种操作：（1）ARRIVE n c表示进货n个，每个c元。（2）BUY n t表示一个买货的人要买n个，一共拿了t元钱。如果现在店里的货的数量大于等于n且最便宜的n个的价格小于等于t则将最便宜的卖给他。否则不卖。

思路：线段树的节点保存该段货物的数量和总价格，另外增加一个标志标记该段内是否被卖完。标记每次向下传递。注意在计算最便宜的n个的价格时向下传递标志时要向上更新sum和value值。因为若这次不向上更新，若本次的价格和大于t则不会再进行删掉前n个的操作，就会出错。

const int N=100005;

const int M=300005;



struct query

{

    char cmd[10];

    i64 n,c;

};



struct node

{

    i64 sum,value;

    int L,R,mid,flag;

};



query q[N];

vector<i64> V;

map<i64,int> Map;

int n,m;

node a[M];



void build(int t,int L,int R)

{

    a[t].L=L;

    a[t].R=R;

    a[t].mid=(L+R)>>1;

    a[t].flag=0;

    a[t].sum=a[t].value=0;

    if(L==R) return;

    build(t*2,L,a[t].mid);

    build(t*2+1,a[t].mid+1,R);

}



void mark(int u)

{

    if(u>M) return;

    a[u].flag=1;

    a[u].sum=a[u].value=0;

}



void down(int t)

{

    if(t>M||!a[t].flag) return;

    a[t].flag=0;

    mark(t*2);mark(t*2+1);

}



void up(int t)

{

    a[t].sum=a[t*2].sum+a[t*2+1].sum;

    a[t].value=a[t*2].value+a[t*2+1].value;

}



void insert(int t,int pos,i64 n)

{

    down(t);

    if(a[t].L==a[t].R)

    {

        a[t].sum+=n;

        a[t].value+=V[pos-1]*n;

        return;

    }

    if(pos<=a[t].mid) insert(t*2,pos,n);

    else insert(t*2+1,pos,n);

    up(t);

}





i64 get(int t,i64 n)

{

    down(t);

    if(n==a[t].sum) return a[t].value;

    if(a[t].L==a[t].R) return a[t].value/a[t].sum*n;

    up(t);

    if(n<=a[t*2].sum) return get(t*2,n);

    return a[t*2].value+get(t*2+1,n-a[t*2].sum);

}



void del(int t,i64 n)

{

    down(t);

    if(n==a[t].sum)

    {

        mark(t);

        return;

    }

    if(a[t].L==a[t].R)

    {

        a[t].value-=a[t].value/a[t].sum*n;

        a[t].sum-=n;

        return;

    }

    if(n<=a[t*2].sum) del(t*2,n);

    else

    {

        del(t*2+1,n-a[t*2].sum);

        mark(t*2);

    }

    up(t);

}



void deal(i64 n,i64 t)

{

    if(a[1].sum<n||get(1,n)>t) puts("UNHAPPY");

    else puts("HAPPY"),del(1,n);

}





int main()

{

    while(scanf("%s%lld%lld",q[n+1].cmd,&q[n+1].n,&q[n+1].c)!=-1)

    {

        n++;

        if(q[n].cmd[0]=='A') V.pb(q[n].c);

    }

    sort(V.begin(),V.end());

    m=unique(V.begin(),V.end())-V.begin();

    int i,k;

    FOR0(i,m) Map[V[i]]=i+1;

    build(1,1,m);

    FOR1(i,n)

    {

        if(q[i].cmd[0]=='A')

        {

            k=Map[q[i].c];

            insert(1,k,q[i].n);

        }

        else deal(q[i].n,q[i].c);

    }

	return 0;

}