Codeforces Round #600 (Div. 2)

B. Silly Mistake

题意
有一个公司,每天有 n 个员工进出,标号1~n。

公司对进入办公室有一些规定:

  • 每个员工每天最多只能进一次办公室。
  • 如果那天他没有进办公室的话,他就没有办法离开。
  • 每天开始和结束时,办公室都是空的(员工不能过夜)。

满足以上三个条件的事件序列被称为有效日。

现给你一个序列 a,序列 a表示了员工的出入情况,“k"代表员工k进入办公室,”-k"代表员工 k 出了办公室。 例如:
[2, -2, 3, -3]
就代表 2 号员工进入了办公室, 2 号员工出了办公室, 3 号员工进入了办公室, 3 号员工出了办公室。

问这个序列 a 是否可以分成若干天的序列,保证每天都是有效的序列,如果无法分配直接输出 -1 ,否则输出可分的天数 和 每天的元素个数。

思路
模拟题
首先考虑非法的情况:

  • 员工在一天进入多次
  • 进入的员工没有出来
  • 没有进员工,就要出员工

下面就来解决非法的情况:
记录序列 a 的前缀和,找出里面前缀和为 0 的情况
前缀和为 0 就说明当前元素到 上一个前缀和为 0 的元素之间的所有所有元素 在不重复的情况下 可以放在一天。

记录每个正数出现的的次数,把出现的次数和前缀和中0 的个数比较,当其次数大于前缀和中的0的时候,意味着 有一个员工多次的进入,有元素重复。

当前缀和小于 0 的时候,意味着有没有进员工,就要出员工的情况存在。

/***********************
*author:ccf
*source:Codeforces_600-B. Silly Mistake
*topic:
************************/
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <vector>
#define ll __int64
using namespace std;

const int N = 1e6 + 7;
int cas,n,cnt = 0;
ll a[N],s[N],bk[N] = {0};
vector<ll> ans;

bool check() {
	int len = ans.size() - 1;
	for(int i = 1; i <= n; i++) {
		if(s[i] == 0) len--;
		if(s[i] < 0)  return true;
		if(a[i] < 0 )  bk[-a[i]]--;
		if(a[i] > 0 && bk[a[i]] > len) return true;
	}
	return false;
}
int main() {
	//freopen("data.in","r",stdin);
	scanf("%d",&n);
	ans.push_back(0);
	ll tmp = 0;
	for(int i = 1; i <= n; i++) {
		scanf("%I64d",&a[i]);
		tmp += a[i];
		if(a[i] > 0)
			bk[a[i]]++;	
		else if(a[i] < 0) {
			if(bk[-a[i]] == 0) {
				printf("-1");
				return 0;
			}
		}
		s[i] = a[i] + s[i-1];
	}
	for(int i = 1; i <= n; i++) {
		if(s[i] == 0) {
			cnt++;
			ans.push_back(i);
		}
	}
	if(check() || tmp != 0 || n % 2 ==1) {
		printf("-1\n");
	} else {
		printf("%d\n",cnt);
		int len = ans.size();
		for(int i = 1; i < len; i++) {
			printf("%I64d ",ans[i]- ans[i-1]);
		}
	}
	return 0;
}
发布了141 篇原创文章 · 获赞 71 · 访问量 6万+

猜你喜欢

转载自blog.csdn.net/sinat_40872274/article/details/103215316