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

256 篇文章 0 订阅
142 篇文章 0 订阅

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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值