[CF286C] Main Sequence

问题描述

定义幸运数列：

空数列是幸运数列

如果 S 是幸运数列，那么 {r, S, -r} 也是幸运数列 (r > 0)

如果 S 和 T 都是幸运数列，那么 {S, T} 也是幸运数列

给定一个幸运数列中每个数的绝对值，并且要求其中的一些数是负数，其他的可正可负。

问是否有合法方案，如果有，给出任意一种方案。 N ≤ 10^6

幸运数列例子：{1, 2, -2, -1, 1, -1, 1, -1}

输入：1 1 1 1 要求第三个数是负数输出： 1 1 -1 -1

样例输入

2

1 1

0

样例输出

YES

1 -1

解析

可以比较容易的看出来这是一个括号序列的形式。如果将正数视为左括号，负数视为右括号，那么给定一些数为负数就说明钦定了一些右括号，需要判断是否有满足要求的左括号。为了限定右括号，我们从右往左扫一遍，如果栈顶元素不等于当前元素或当前元素被指定为负数，就将元素入栈并标记为负数；否则弹出栈顶元素。最后如果栈不为空，说明无解；否则输出答案。

代码

#include <iostream>

#include <cstdio>

#define int long long

#define N 1000002

using namespace std;

int n,m,i,a[N],q[N],s[N],top;

bool f[N];

int read()

{

	char c=getchar();

	int w=0;

	while(c<'0'||c>'9') c=getchar();

	while(c<='9'&&c>='0'){

		w=w*10+c-'0';

		c=getchar();

	}

	return w;

}

signed main()

{

	n=read();

	for(i=1;i<=n;i++) a[i]=read();

	m=read();

	for(i=1;i<=m;i++){

		q[i]=read();

		f[q[i]]=1;

	}

	for(i=n;i>=1;i--){

		if(s[top]!=a[i]||f[i]) s[++top]=a[i],f[i]=1;

		else top--;

	}

	if(top){

		puts("NO");

		return 0;

	}

	puts("YES");

	for(i=1;i<=n;i++){

		if(f[i]) printf("%lld ",-a[i]);

		else printf("%lld ",a[i]);

	}

	puts("");

	return 0;

}

反思

若将原序列视为括号序列，则给定哪些数是负数就相当于钦定了一些右括号的位置。所以从右往左会比从左往右少写很多东西。还是要注意想好之后能否简化代码的写法，否则有可能会带来非常多的麻烦。

巴特西

[CF286C] Main Sequence

问题描述

样例输入

样例输出

解析

代码

反思

最新文章

热门文章