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