HDU 4614 Vases and Flowers

HDU 4614 Vases and Flowers

线段树

题意

两个数n,m, n表示花瓶数量0~n-1，开始全为空。接下来有m组，每组有数k，a，b;当k=1时， 从a位置开始插花，b为花数量，当花瓶有花时，跳过当前，直到找到空花瓶再插入，输出插入第一支花的位置和最后一支花的位置，花可以没插完，当一支都没有插入则输出Can not put any one.；当k=2时，清空【a,b】的花瓶，并输出在被清空的花的数量。

思路

线段树用1表示空位，0表示有花。
清空实质就是统计a,b之间有多少花，并把区间a,b强制置1。
插花要有点技巧，设置两个查询函数：查询l,r之间空位的数量，查询从第开头数p个空位到哪。
这样从a开始插b朵花：先查询a到末尾与多少空位x，若x==0输出没法插花；x小于b就把b换成x，实际插x朵；否则插b朵。然后查询开头到a有多少空位k，从开头数k+1个空位到哪，数k+b个空位到哪。这两个数就是实际插花的左右端点。在将这个区间全置0即可。

应该有线段树上二分的写法，可以直接查询从a往右数b个空瓶到哪，可是我不会写。

代码

注意题目从0开始编号，注意转换与边界处理。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<set>
#include<iostream>
#include<vector>
#include<queue>
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define M(a,b) memset(a,b,sizeof(a))
using namespace std;
const int MAXN=50005;
const int oo=0x3f3f3f3f;
typedef long long LL;
struct Stree
{
    int sum, lpos, rpos;
    int set;
    Stree() { sum=lpos=rpos=0;set=-1; }
    Stree(int _a, int _b, int _c) { sum=_a, lpos=_b, rpos=_c; }
}stree[MAXN<<2];
void pushup(int rt)
{
    stree[rt].sum=stree[rt<<1].sum+stree[rt<<1|1].sum;
    stree[rt].lpos=(stree[rt<<1].lpos==-1 ? stree[rt<<1|1].lpos : stree[rt<<1].lpos);
    stree[rt].rpos=(stree[rt<<1|1].rpos==-1 ? stree[rt<<1].rpos : stree[rt<<1|1].rpos);
}
void build(int l, int r, int rt)
{
    stree[rt].lpos=stree[rt].rpos=stree[rt].set=-1;
    stree[rt].sum=0;
    if(l==r)
    {
        stree[rt].lpos=stree[rt].rpos=l;
        stree[rt].sum=1;
        return;
    }
    int mid=(l+r)>>1;
    build(lson), build(rson);
    pushup(rt);
}
void pushdown(int l, int r, int rt)
{
    if(stree[rt].set!=-1)
    {
        int c=stree[rt].set;
        stree[rt<<1].set=stree[rt<<1|1].set=c;stree[rt].set=-1;
        int mid=(l+r)>>1;
        if(c==0)
        {
            stree[rt<<1].lpos=stree[rt<<1].rpos=stree[rt<<1|1].lpos=stree[rt<<1|1].rpos=-1;
            stree[rt<<1].sum=stree[rt<<1|1].sum=0;
        }
        else
        {
            stree[rt<<1].lpos=l;
            stree[rt<<1].rpos=mid;
            stree[rt<<1|1].lpos=mid+1;
            stree[rt<<1|1].rpos=r;
            stree[rt<<1].sum=mid-l+1;
            stree[rt<<1|1].sum=r-mid;
        }
    }
}
int update(int L, int R, int col, int l, int r, int rt)
{
    if(L<=l&&r<=R)
    {
        if(col==1)
        {
            int c=stree[rt].sum;
            c=r-l+1-c;
            stree[rt].sum=r-l+1;
            stree[rt].set=1;
            stree[rt].lpos=l, stree[rt].rpos=r;
            return c;
        }
        else
        {
            int c=stree[rt].sum;
            stree[rt].sum=0;
            stree[rt].set=0;
            stree[rt].lpos=-1, stree[rt].rpos=-1;
            return c;
        }
    }
    int mid=(l+r)>>1;
    pushdown(l, r, rt);
    int res=0;
    if(mid<L) res=update(L, R, col, rson);
    else if(R<=mid) res=update(L, R, col, lson);
    else res=update(L, R, col, lson)+update(L, R, col, rson);
    pushup(rt);
    return res;
}
int query(int L, int R, int l, int r, int rt)
{
    if(L<=l&&r<=R) return stree[rt].sum;
    pushdown(l, r, rt);
    int mid=(l+r)>>1;
    if(R<=mid) return query(L, R, lson);
    else if(mid<L) return query(L, R, rson);
    else return query(L, R, lson)+query(L, R, rson);
}
int query_pos(int p, int l, int r, int rt)
{
    if(l==r) return l;
    else if(stree[rt].sum==p) return stree[rt].rpos;
    int mid=(l+r)>>1;
    pushdown(l, r, rt);
    if(stree[rt<<1].sum>=p)
        return query_pos(p, lson);
    else 
        return query_pos(p-stree[rt<<1].sum, rson);
}

int main()
{
    int T;scanf("%d", &T);
    while(T--)
    {
        int n, m;scanf("%d%d", &n, &m);
        build(1, n, 1);
        while(m--)
        {
            int a, b, c;scanf("%d%d%d", &a, &b, &c);
            if(a==1)
            {
                b++;
                int sum=query(b, n, 1, n, 1);
                if(sum==0) puts("Can not put any one.");
                else
                {
                    int pre=(b==1?0:query(1, b-1, 1, n, 1));
                    c=min(c, sum);
                    int l=query_pos(pre+1, 1, n, 1);
                    int r=query_pos(pre+c, 1, n, 1);
                    update(l, r, 0, 1, n, 1);
                    printf("%d %d\n", l-1, r-1);
                }
            }
            else
            {
                printf("%d\n", update(b+1, c+1, 1, 1, n, 1));
            }
        }
        printf("\n");
    }
    //system("pause");
    return 0;
}