codeforce 940D 思维题

传送门：http://codeforces.com/problemset/problem/940/D

题意：

b1 = b2 = b3 = b4 = 0.
For all 5 ≤ i ≤ n:
bi = 0 if ai, ai - 1, ai - 2, ai - 3, ai - 4 > r and bi - 1 = bi - 2 = bi - 3 = bi - 4 = 1
bi = 1 if ai, ai - 1, ai - 2, ai - 3, ai - 4 < l and bi - 1 = bi - 2 = bi - 3 = bi - 4 = 0
bi = bi - 1 otherwise
对于给定的a数组和b数组，找出其中一个l，r使得上述的条件满足。

分析：

对于l，r有约束作用的只有以下的四种情况
（1）$b_{i<=j<=i+4}$=00000
（2）$b_{i<=j<=i+4}$=00001
（3）$b_{i<=j<=i+4}$=11111
（4）$b_{i<=j<=i+4}$=11110
对于（1）$l<=MAX_{i<=j<=i+4}a_j$
对于（2）$l>=MAX_{i<=j<=i+4}a_j+1$
对于（3）$r>=MIN_{i<=j<=i+4}a_j$
对于（4）$l>=MIN_{i<=j<=i+4}a_j-1$

代码：

#include <iostream>
#include <bits/stdc++.h>
#define INF 1000000000
using namespace std;

char b[100005];
int a[100005];
int main()
{
    int n;
    cin>>n;
    for(int i=0;i<n;i++)
    {
        scanf("%d",&a[i]);
    }
    scanf("%s",b);
    int x=0;         //用于表示前5为数位
    int l=-INF,r=INF;
    for(int i=4;i<n;i++)
    {
        x<<=1;
        x+=b[i]-'0';
        x%=32;
        if(x==0)
        {
            int ma=-INF;
            for(int j=i;j>=i-4;j--)
            {
                ma=max(ma,a[j]);
            }
            l=min(l,ma);
        }else if(x==1){
            int ma=-INF;
            for(int j=i;j>=i-4;j--)
            {
                ma=max(ma,a[j]+1);
            }
            l=max(l,ma);
        }else if(x==31){
            int mi=INF;
            for(int j=i;j>=i-4;j--)
            {
                mi=min(mi,a[j]);
            }
            r=max(r,mi);
        }else if(x==30){
            int mi=INF;
            for(int j=i;j>=i-4;j--)
            {
                mi=min(mi,a[j]-1);
            }
            r=min(r,mi);
        }

    }
    printf("%d %d",l,r);
    return 0;
}