Codeforces Round #529 (Div. 3) E. Almost Regular Bracket Sequence

题目大意就是给一个长度为n括号序列,然后问你有几个位置,你将该位置的括号反转之后,这个括号序列可以变成合法的括号序列。合法的括号序列定义就是 1.括号序列里左右括号数量相等 2.对于i位置对应的前缀里左括号数量>=右括号数量(1<=i<=n)

先设左括号为1,右括号为-1,然后求出括号序列对应的前缀和数组a

假设要将pos位置的左括号改成右括号是满足条件的。

那么必有

              1.未修改前左括号个数比右括号个数多2

              2.未修改前 a[i]>=0  (1<=i<=pos-1),可以等效成min(a[i] (1<=i<=pos-1))>=0

              3.修改之后 a[i]>=0  (pos<=i<=n)

仅仅用这三个条件是无法在题目要求时间内求出答案的,因为第3个条件的判断需要把a[i] (pos<=i<=n)都减2然后再判断是否全都>=0,这样的话时间复杂度很高,所以这里需要优化,因为把pos位置的左括号变成右括号,会对pos位置的前缀值造成-2的作用,当然pos+1到n位置的前缀因为包含了pos位置,所以pos+1到n位置的前缀值也都会-2,总的来说就是a[i]会减小2  (pos<=i<=n)

,然后要求减小后>=0,那么就要a[i]>=2,然后是pos到n范围的前缀值a都满足,那么就等效成min(a[i]  (pos<=i<=n))>=2.

这样判断pos位置的括号能否反转就仅仅和原数组a有关了,我们只需预处理出前缀数组a的前缀最小值数组premin,和前缀数组a的后缀最小值数组sufmin,就可以对pos位置O(1)判断了,因此总复杂度O(N)

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+10;
int a[maxn],premin[maxn],sufmin[maxn];
char s[maxn];
int main()
{
    int n,tmp=0,ans=0,low=0;
    scanf("%d",&n);
    premin[0]=sufmin[n+1]=maxn;
    for(int i=1;i<=n;i++)
    {
        scanf(" %c",&(s[i]));
        if(s[i]=='(')
            tmp+=1;
        else
            tmp-=1;
        a[i]=tmp;
        premin[i]=min(a[i],premin[i-1]);
    }
    for(int i=n;i>=1;i--)
    {
        sufmin[i]=min(a[i],sufmin[i+1]);
    }
    if(n&1||(tmp!=-2&&tmp!=2))
    {
        printf("0\n");
        return 0;
    }
    /*
    for(int i=1;i<=n;i++)
        cout<<a[i]<<"   ";
    cout<<endl<<endl;
    for(int i=1;i<=n;i++)
        cout<<premin[i]<<"  ";
    cout<<endl<<endl;
    for(int i=1;i<=n;i++)
        cout<<sufmin[i]<<"  ";
    cout<<endl<<endl;
    */
    if(tmp==2)
        low=2;
    else
        low=-2;
    for(int i=1;i<=n;i++)
    {
        if(tmp==2&&s[i]=='(')
        {
            if(premin[i]>=0&&sufmin[i]>=low)
                ans++;
        }
        else
        {
            if(tmp==-2&&s[i]==')')
            {
                if(i==1&&sufmin[i]>=low)
                    ans++;
                if(i!=1&&premin[i-1]>=0&&sufmin[i]>=low)
                    ans++;
            }
        }
    }
    printf("%d\n",ans);
    return 0;
}

 

posted @ 2019-01-01 22:58  eason99  阅读(73)  评论(0编辑  收藏  举报