ZOJ - 2301- Color the Ball(线段树区间染色+离散化)

题目:ZOJ - 2301

题意:有无数多个球,每个球最初被染成了黑色,有n个操作,把从l到r的球染成黑色或者白色,问n个操作后,统计哪段区间白色最长。

题解:这个题目跟POJ - 2528很相似,推荐先去做做这个题目,离散化操作,以离散化后的数组下标作为线段树的建树区间,但是这里要注意一下,不同于POJ-2528,那个题目要求的是最终有多少种颜色,不精确到具体区间,那么我只需要在间距大于1的地方补一个数就可以,但是这个题目要求精确到具体区间,我在左边区间那里补一个数还不够,比如说:8 11 w这里被染成白色,如果我只增加一个数就变成了8 9 11,那么当我11 11 b时,我的答案区间应该是8-10,但是在这里就会变成8-9了,所以我需要在右边再补一个数10,这样不论左右两边的两个位置怎么变,我的答案总会是对的了。

代码:

#include
#define N 2005
#define L node<<1
#define R node<<1|1
using namespace std;
int n,b[N*2],m,sum[N<<2],ans,ansl,ansr;
bool vis[N<<2];
struct xqy
{
	int ll,rr,c;
}a[N];
void pushdown(int node)
{
	if(sum[node]!=-1)
	{
		sum[L]=sum[node];
		sum[R]=sum[node];
		sum[node]=-1;
	}
}
void update(int node,int l,int r,int x,int y,int z)
{
	if(x<=l&&r<=y)
	{
		sum[node]=z;
		return ;
	}
	pushdown(node);
	int mid=(l+r)>>1;
	if(x<=mid)update(L,l,mid,x,y,z);
	if(y>mid)update(R,mid+1,r,x,y,z);
}
void query(int node,int l,int r)
{
	if(sum[node]==1)
	{
		for(int i=l;i<=r;i++)
			vis[i]=1;
		return ;
	}
	if(sum[node]==0)return ;
	pushdown(node);
	int mid=(l+r)>>1;
	query(L,l,mid);
	query(R,mid+1,r);
}
int main()
{
	while(~scanf("%d",&n))
	{
		int tot=0;
		for(int i=1;i<=n;i++)
		{
			int x,y;char cc;
			scanf("%d%d %c",&x,&y,&cc);
			a[i].ll=x;
			a[i].rr=y;
			if(cc=='w')a[i].c=1;
			else a[i].c=0;
			b[++tot]=x;
			b[++tot]=y;
		}
		sort(b+1,b+tot+1);
		int num=unique(b+1,b+tot+1)-(b+1);
		m=num;
		for(int i=1;i1)
			{
				b[++m]=b[i]+1;
				if(b[i+1]-b[i]>2)
					b[++m]=b[i+1]-1;//离散化操作必须要这样
			}
		sort(b+1,b+m+1);
		memset(sum,-1,sizeof(sum));
		for(int i=1;i<=n;i++)
		{
			int x=lower_bound(b+1,b+m+1,a[i].ll)-b;
			int y=lower_bound(b+1,b+m+1,a[i].rr)-b;
			update(1,1,m,x,y,a[i].c);
		}
		query(1,1,m);
		ans=-1;
		for(int i=1;i<=m;i++)//寻找最长的区间
		{
			if(vis[i])
			{
				int j;
				for(j=i;j<=m;j++)
					if(!vis[j])break;
				if(b[j-1]-b[i]>ans)
				{
					ans=b[j-1]-b[i];
					ansr=b[j-1];
					ansl=b[i];
				}
				i=j;
			}
		}
		if(ans==-1)printf("Oh, my god\n");
		else printf("%d %d\n",ansl,ansr);
	}
	return 0;
}

 

你可能感兴趣的:(ACM_刷题,ACM_线段树)