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