首页 > 代码库 > EOJ 3246 实验室传染病

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;}

EOJ 3246 实验室传染病