codeforces 940D. Alena And The Heater （罗辑思维）

传送门：codeforces 940D

题目大意：

数组 b 根据数组 a 和 l,r 的值来生成。有以下规则：

规则1：如果 a[i]、a[i-1]、a[i-2]、a[i-3]、a[i-4] 都 >r 并且 b[i-1]=b[i-2]=b[i-3]=b[i-4]=1，则 b[i] = 0

规则2：如果 a[i]、a[i-1]、a[i-2]、a[i-3]、a[i-4] 都 <l 并且 b[i-1]=b[i-2]=b[i-3]=b[i-4]=0，则 b[i] = 1

规则3：否则 b[i] = b[i-1]

现在，已知数组 a、b，让你求 l、r 的值。

思路：

其实就是要找一个分类讨论的标准，让所有的可能不重叠。大体的思路是先将 l 设为最小值，将 r 设为最大值，然后根据条件缩小区间的范围。

我们分类的标准是 b[i-1]=b[i-2]=b[i-3]=b[i-4]=1，b[i-1]=b[i-2]=b[i-3]=b[i-4]=0 和 b[i-1]，b[i-2]，b[i-3]，b[i-4]中既有 0 又有 1。

既有 0 又有 1 时规则1 和规则2 都不满足，采用规则 3。不涉及 l 和 r 的值。全为 1 时，如果当前位 b[i]=0，则 a[i]、a[i-1]、a[i-2]、a[i-3]、a[i-4] 都 >r，也就是说 a[i]、a[i-1]、a[i-2]、a[i-3]、a[i-4] 中的最小值要 > r；如果当前位 b[i]=1，采用的是规则 3，所以 a[i]、a[i-1]、a[i-2]、a[i-3]、a[i-4] 中至少有一个数 <= r ，即它们的最小值应该 <=r。全为 0 时和全为 1 的情况类似。

代码：

#include<stdio.h>
#include<iostream>
#define inf 1e9
using namespace std;

int a[100010];

int getmax(int l,int r)
{
	int i,ans=-inf;
	for(i=l;i<=r;i++)
		if(a[i]>ans) ans=a[i];
	return ans;
}

int getmin(int l,int r)
{
	int i,ans=inf;
	for(i=l;i<=r;i++)
		if(a[i]<ans) ans=a[i];
	return ans;
}

int main()
{
	int i,j,n,l,r,f1,f2,Max,Min;
	char b[100010];
	while(~scanf("%d",&n))
	{
		for(i=1;i<=n;i++) scanf("%d",&a[i]);
		scanf("%s",b+1);		
		l=-inf;
		r=inf;
		for(i=5;i<=n;i++)
		{
			if(b[i-1]=='1'&&b[i-2]=='1'&&b[i-3]=='1'&&b[i-4]=='1')
			{ //全为 1
				Min=getmin(i-4,i); //区间最小值 
				if(b[i]=='0') r=min(r,Min-1); //规则 1 
				else r=max(r,Min); //规则 3 
			}
			else if(b[i-1]=='0'&&b[i-2]=='0'&&b[i-3]=='0'&&b[i-4]=='0')
			{ //全为 0 
				Max=getmax(i-4,i); //区间最大值 
				if(b[i]=='1') l=max(l,Max+1); //规则 2 
				else l=min(l,Max); //规则 3 
			}
		}
		printf("%d %d\n",l,r);
	}
	return 0;
}