EOJ 3246 实验室传染病

线段树,暴力。

先处理出每个点直接能感染到的最左边的和最右边的。

之后每次扩展,看向左能到达的那些点中,最左以及最右能到哪些点,更新。

看向右能到达的那些点中,最左以及最右能到哪些点,更新。

最左最右能到哪些点,可以用线段树计算区间最小值最大值。

如果某次无法更新,那么结束。

#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
using namespace std;

int n;
struct X
{
	int x,y;
	int ll,rr;
	int id;
}s[100010];

int L[100010], R[100010], ans[100010];

bool cmp(X a,X b)
{
	return a.x<b.x;
}

int mn[2][4*100010];
int mx[2][4*100010];

int B;

void update(int op,int pos,int val,int l,int r,int rt)
{
	if(l==r)
	{
		mn[op][rt] = mx[op][rt] = val;
		return ;
	}

	int mid = ( l + r) / 2;
	if(pos<=mid) update(op,pos,val,l,mid,2*rt);
	else update(op,pos,val,mid+1,r,2*rt+1);
	
	mn[op][rt] = min(mn[op][2*rt], mn[op][2*rt+1]);
	mx[op][rt] = max(mx[op][2*rt], mx[op][2*rt+1]);
}

void MIN(int op,int L,int R,int l,int r,int rt)
{
	if(L<=l&&r<=R)
	{
		B = min(B,mn[op][rt]);
		return ;
	}

	int mid = ( l + r) / 2;
	if(L<=mid) MIN(op,L,R,l,mid,2*rt);
	if(R>mid) MIN(op,L,R,mid+1,r,2*rt+1);
}

void MAX(int op,int L,int R,int l,int r,int rt)
{
	if(L<=l&&r<=R)
	{
		B = max(B,mx[op][rt]);
		return ;
	}

	int mid = ( l + r) / 2;
	if(L<=mid) MAX(op,L,R,l,mid,2*rt);
	if(R>mid) MAX(op,L,R,mid+1,r,2*rt+1);
}

int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++) 
	{
		scanf("%d%d",&s[i].x,&s[i].y);
		s[i].ll = s[i].x - s[i].y;
		s[i].rr = s[i].x + s[i].y;
		s[i].id = i;
	}

	sort(s+1,s+1+n,cmp);
	
	for(int i=1;i<=n;i++) 
	{
		int left = 1, right = i, pos;
		while(left <= right)
		{
			int mid = (left + right)/2;
			if(s[mid].x >= s[i].ll) pos = mid, right = mid-1;
			else left = mid + 1;
		}

		L[i] = pos;

		left = i, right = n;
		while(left<=right)
		{
			int mid = (left + right)/2;
			if(s[mid].x <= s[i].rr) pos = mid, left = mid+1;
			else right = mid - 1;
		}

		R[i] = pos;
	}

	for(int i=1;i<=n;i++)
	{
		update(0,i,L[i],1,n,1);
		update(1,i,R[i],1,n,1);
	}

	while(1)
	{
		bool f = 0;

		for(int i=1;i<=n;i++)
		{
			B = n+1;
			MIN(0,L[i],i,1,n,1);
			if(B != L[i]) f=1, L[i] = B, update(0,i,L[i],1,n,1);
		}

		for(int i=n;i>=1;i--)
		{
			B = 0;
			MAX(1,i,R[i],1,n,1);
			if(B != R[i]) f=1, R[i] = B, update(1,i,R[i],1,n,1);
		}

		for(int i=1;i<=n;i++)
		{
			B = 0;
			MAX(1,L[i],i,1,n,1);
			if(B != R[i]) f=1, R[i] = B, update(1,i,R[i],1,n,1);
		}

		for(int i=n;i>=1;i--)
		{
			B = n+1;
			MIN(0,i,R[i],1,n,1);
			if(B != L[i]) f=1, L[i] = B, update(0,i,L[i],1,n,1);
		}
		
		if( f == 0 ) break;
	}

	for(int i=1;i<=n;i++) ans[s[i].id] = R[i] - L[i] + 1;

	for(int i=1;i<=n;i++)
	{
		printf("%d",ans[i]);
		if(i<n) printf(" ");
		else printf("\n");
	}

	return 0;
}
posted @ 2017-05-13 20:45  Fighting_Heart  阅读(289)  评论(0编辑  收藏  举报