CodeForces 287 E.Main Sequence(贪心+栈)

来源:互联网 发布:小布老师linux 编辑:程序博客网 时间:2024/05/16 07:01

Description
定义由整数组成的括号序列,只有一下三种序列是括号序列:
1. 空序列
2. 两个括号序列放一起
3. 一个括号序列左边放个v,右边放个-v(v > 0)
现在给出一个全部是正整数的序列,给出m个编号表示这些编号对应的值是负数,其他值的符号随意,问是否有一种合适的赋符号的方案使得该序列是括号序列
Input
第一行一整数n表示该序列长度,之后输入n个正整数x[i]表示该序列,然后输入一整数m和m个整数表示x序列中一些负数的编号(1<=n<=1e6,1<=x[i]<=1e9,0<=m<=n)
Output
如果存在合法方案则输出YES和赋予符号后的括号序列,否则输出NO
Sample Input
4
1 1 1 1
1 3
Sample Output
YES
1 1 -1 -1
Solution
从后往前看,用一个栈把负数全部存起来,如果当前的数符号确定(负)就进栈,如果符号不确定就看这个数加上栈顶元素是不是0,如果是0说明这个数和栈顶元素可以匹配(贪心,可以匹配就匹配,因为如果当前不匹配,就要从前面找两个数与这两个数匹配,失败的几率更大;而且如果可以从前面找两个数和这两个数匹配,那么前两个数匹配,这两个数匹配也是可以的),栈顶元素出栈,否则不能匹配说明这个数只能是负数,扫一遍后如果栈空说明每个元素都匹配了,有解,否则无解
Code

#include<cstdio>#include<iostream>#include<cstring>#include<algorithm>#include<cmath>#include<vector>#include<queue>#include<map>#include<set>#include<ctime>using namespace std;typedef long long ll;#define INF 0x3f3f3f3f#define maxn 1111111int n,x[maxn],s[maxn];int main(){    while(~scanf("%d",&n))    {        for(int i=1;i<=n;i++)scanf("%d",&x[i]);        int m,q,p=0;        scanf("%d",&m);        while(m--)        {            scanf("%d",&q);            x[q]*=-1;        }        p=0;        for(int i=n;i>=1;i--)        {            if(x[i]<0)s[++p]=x[i];            else            {                if(p&&x[i]+s[p]==0)p--;                else s[++p]=-x[i],x[i]*=-1;            }        }        if(p)printf("NO\n");        else         {            printf("YES\n");            for(int i=1;i<=n;i++)printf("%d%c",x[i],i==n?'\n':' ');        }    }    return 0;}