HDU 4614

给定一个区间[0,N-1],初始时每个位置上的数字都是0，可以对其进行以下两种操作：

1、在位置A开始寻找F(如果没有这么多，则有多少个就找多少个)个数值为0的位置，把位置上的数修改为1,并返回第一个和最后一个修改的位置

2、查询区间[a,b]内1的个数，并把区间[a,b]每个位置上的数修改为0

很明显

对于线段树，保存区间和就可以了。

对于二还是很好处理的，直接区间长度 - 去区间内0的个数，并且更新一下

而对于一，如果暴力的话就是从左到右遍历一遍，但是不够优秀，既然能遍历，那就能二分，二分找出0~L的num个0，再找出0~R的num+y个0，这区间就是我们要填 0 的区间了。

以下是 AC 代码

#include <cstdio>
#include <cstring>
#include <iostream>
#define LL long long
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
using namespace std;
const int maxn = 50010;
int sum[maxn<<2],add[maxn<<2];
void pushup(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void build(int l,int r,int rt)
{
    add[rt]=-1;
    if(l==r)
    {
        sum[rt]=1;
        return;
    }
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(rt);
}
void pushdown(int rt,int len)
{
    if(add[rt]!=-1)
    {
        add[rt<<1]=add[rt<<1|1]=add[rt];
        sum[rt<<1]=(len-(len>>1))*add[rt];
        sum[rt<<1|1]=(len>>1)*add[rt];
        add[rt]=-1;
    }
}
void update(int L,int R,int c,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        add[rt]=c;
        sum[rt]=(r-l+1)*c;
        return;
    }
    pushdown(rt,r-l+1);
    int m=(l+r)>>1;
    if(L<=m)
        update(L,R,c,lson);
    if(m<R)
        update(L,R,c,rson);
    pushup(rt);
}
int query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        return sum[rt];
    }
    pushdown(rt,r-l+1);
    int m=(l+r)>>1;
    int res=0;
    if(L<=m)
        res+=query(L,R,lson);
    if(m<R)
        res+=query(L,R,rson);
    return res;
}
int bin(int x,int y,int num,int cnt)
{
    int l=x,r=y,ans;
    while(l<=r)
    {
        int mid=(l+r)>>1;
        if(query(0,mid,0,y,1)-num >= cnt)
            r=mid-1,ans=mid;
        else 
            l=mid+1;
    }
    return ans;
}
int main()
{
    int t,n,m,num;
    int op,x,y;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        build(0,n-1,1);
        while(m--)
        {
            scanf("%d%d%d",&op,&x,&y);
            if(op==1)
            {
                int sum=query(x,n-1,0,n-1,1);
                if(sum==0)
                {
                    puts("Can not put any one.");
                    continue;
                }
                if(sum<y)
                    y=sum;
                if(x-1>=0)
                    num=query(0,x-1,0,n-1,1);
                else 
                    num=0;
                int left,right;
                left=bin(x,n-1,num,1);
                right=bin(x,n-1,num,y);
                printf("%d %d\n",left,right);
                update(left,right,0,0,n-1,1);
            }
            else
            {
                printf("%d\n",y-x+1-query(x,y,0,n-1,1));
                update(x,y,1,0,n-1,1);
            }
        }
        puts("");
    }
    return 0;
}

猜你喜欢