hdu 4614 线段树+二分

题目

这题我们的线段树储存的是区间内空的树有几个,这样就很好做了。如果区间内没有空的那肯定输出那个字符串,然后二分去找k=1的l和r,k=2就用区间长度减去区间询问就好了,其实也不难。

#include
#include
#include
#include
#include
#include
#include 
#include
#include
#include 
#include
#include
#include
#include
#include
#define IO ios::sync_with_stdio(false)
#define eps 1e-7
#define int long long
using namespace std;
int tt,n,m,k;
struct node
{
	int l,r,sum,add;
}t[50005*4];
void build(int rt,int l,int r)
{
	t[rt].l=l,t[rt].r=r;
	t[rt].sum=r-l+1;
	t[rt].add=-1;
	if(l==r)
	return;
	int mid=l+r>>1;
	build(rt*2,l,mid);
	build(rt*2+1,mid+1,r);
	t[rt].sum=t[rt*2].sum+t[rt*2+1].sum;
}
void spread(int rt)
{
	if(t[rt].add!=-1)
	{
		t[rt*2].add=t[rt].add;
		t[rt*2+1].add=t[rt].add;
		t[rt*2].sum=(t[rt*2].r-t[rt*2].l+1)*t[rt].add;
		t[rt*2+1].sum=(t[rt*2+1].r-t[rt*2+1].l+1)*t[rt].add;
		t[rt].add=-1;
	}
}
void upd(int rt ,int l,int r,int d)
{
	if(l<=t[rt].l&&r>=t[rt].r)
	{
		t[rt].sum=(t[rt].r-t[rt].l+1)*d;
		t[rt].add=d;
		return;
	}
	spread(rt);
	int mid=t[rt].l+t[rt].r>>1;
	if(l<=mid)
	{
		upd(rt*2,l,r,d);
	}
	if(r>mid)
	{
		upd(rt*2+1,l,r,d);
	}
	t[rt].sum=t[rt*2].sum+t[rt*2+1].sum;
}
int ask(int rt,int l,int r)
{
	if(l<=t[rt].l&&r>=t[rt].r)
	{
		return t[rt].sum;
	}
	spread(rt);
	int mid=t[rt].l+t[rt].r>>1;
	if(r<=mid)
	{
		return ask(rt*2,l,r);
	}
	else if(l>mid)
	{
		return ask(rt*2+1,l,r);
	}
	else
	{
		return ask(rt*2,l,r)+ask(rt*2+1,l,r);
	}
}
int erfen(int pos,int num)
{
	int l=pos,r=n;
	while(l<r)
	{
		int mid=l+r>>1;
		if(ask(1,pos,mid)>=num)
		{
			r=mid;
		}
		else
		{
			l=mid+1;
		}
	}
	return l;
}
signed main()
{
	IO;
	cin>>tt;
	while(tt--)
	{
		cin>>n>>m;
		build(1,1,n);
		for(int i=1;i<=m;i++)
		{
			cin>>k;
			if(k==1)
			{
				int a,f;
				cin>>a>>f;
				a++;
				int cnt=ask(1,a,n);
				if(cnt==0)
				{
					cout<<"Can not put any one."<<endl;
				}
				else
				{
					int l=erfen(a,1);
					int r=erfen(a,min(cnt,f));
					upd(1,l,r,0);
					cout<<l-1<<" "<<r-1<<endl;
				}
			}
			else
			{
				int l,r;
				cin>>l>>r;
				l++,r++;
				cout<<r-l+1-ask(1,l,r)<<endl;
				upd(1,l,r,1);
			}
		}
		cout<<endl;
	}
}

你可能感兴趣的:(线段树,二分)