HDU - 4614 Vases and Flowers(二分+区间修改)

https://cn.vjudge.net/problem/HDU-4614

题意

n个花瓶，m个操作，花瓶里面有的有花，有的是空的。1操作是从a开始往右放b朵花，花瓶有了的不放，跳过，直到a右边都放满了花，多余的扔了。输出本次放花的起始位置，如果一朵不能放，输出一句话。

分析

1~N 的区间，用1表示空的，0表示放了花的。维护一个sum,就是和。对于操作二就是求区间和。

操作一用二分找第一个符合位置的数。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>
#include <algorithm>
#include <cmath>
#include <ctime>
#include <vector>
#include <queue>
#include <map>
#include <stack>
#include <set>
#include <bitset>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
#define ms(a, b) memset(a, b, sizeof(a))
#define pb push_back
#define mp make_pair
#define pii pair<int, int>
#define eps 0.0000000001
#define IOS ios::sync_with_stdio(0);cin.tie(0);
#define random(a, b) rand()*rand()%(b-a+1)+a
#define pi acos(-1)
const ll INF = 0x3f3f3f3f3f3f3f3fll;
const int inf = 0x3f3f3f3f;
const int maxn = 1e5 + ;
const int maxm =  + ;
const int mod = ;

struct ND{
    int l,r;
    int sum,lazy;
}tree[maxn<<];
int n,m;
void build(int rt,int l,int r){
    tree[rt].lazy=-;
    tree[rt].sum=;
    tree[rt].l=l,tree[rt].r=r;
    if(l==r) return;
    int mid = (l+r)>>;
    build(rt<<,l,mid);
    build(rt<<|,mid+,r);
}
void pushup(int rt){
    tree[rt].sum=tree[rt<<].sum+tree[rt<<|].sum;
}
void pushdown(int rt){
    tree[rt<<].lazy=tree[rt<<|].lazy=tree[rt].lazy;
    if(tree[rt].lazy){
        int len=tree[rt].r-tree[rt].l+;
        tree[rt<<].sum=len-(len>>);
        tree[rt<<|].sum=(len>>);
    }else{
        tree[rt<<].sum=tree[rt<<|].sum=;
    }
    tree[rt].lazy=-;
}
void update(int rt,int l,int r,int val){
    if(l==tree[rt].l&&tree[rt].r==r){
        tree[rt].lazy=val;
        if(val) tree[rt].sum=tree[rt].r-tree[rt].l+;
        else tree[rt].sum=;
        return;
    }
    if(tree[rt].lazy!=-) pushdown(rt);
    int mid = (tree[rt].l+tree[rt].r)>>;
    if(r<=mid) update(rt<<,l,r,val);
    else if(l>mid) update(rt<<|,l,r,val);
    else{
        update(rt<<,l,mid,val);
        update(rt<<|,mid+,r,val);
    }
    pushup(rt);
}
int query(int rt,int l,int r){
    if(tree[rt].l==l&&tree[rt].r==r) return tree[rt].sum;
    if(tree[rt].lazy!=-) pushdown(rt);
    int mid = (tree[rt].l+tree[rt].r)>>;
    if(r<=mid) return query(rt<<,l,r);
    else if(l>mid) return query(rt<<|,l,r);
    else return query(rt<<,l,mid)+query(rt<<|,mid+,r);
}
int bin(int s,int ra){//二分找第一个可以插ra朵花的位置
    int l=s,r=n;
    int ans=-;
    int mid;
    while(l<=r){
        mid=(l+r)>>;
        int tmp=query(,s,mid);
        if(tmp+ra==mid-s+){
            ans=mid;
            r=mid-;
        }else{
            if(tmp+ra<mid-s+) r=mid-;
            else l=mid+;
        }
    }
    return ans;
}
int main() {
#ifdef LOCAL
    freopen("in.txt", "r", stdin);
//    freopen("output.txt", "w", stdout);
#endif
    int t,cas=;
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&n,&m);
        build(,,n);
//        printf("Case #%d:\n",cas++);
        int x,y,c;
        while(m--){
            scanf("%d%d%d",&c,&x,&y);
            if(c==){
                int st,ed;
                x++;
                st=bin(x,);
                if(st==-){
                    puts("Can not put any one.");
                }else{
                    int tmp=query(,st,n);
                    tmp=n-st+-tmp; //最多能插几朵花
                    if(tmp<=y) y=tmp;
                    ed=bin(x,y);
                    printf("%d %d\n",st-,ed-);
                    update(,st,ed,);
                }
            }else{
                x++,y++;
                printf("%d\n",query(,x,y));
                update(,x,y,);
            }
        }
        puts("");
    }
    return ;
}