题目链接:https://codeforces.com/contest/1144/problem/G
题意:现在有一个数列a,问是否可以将这个数列分成两个子序列,两个子序列分别严格递增和严格递减。如果不可以输出no,否则每个元素的归属,元素属于递增子序列标号0,属于递减子序列标号1。
解题心得:就是一个贪心题,现在维护两个序列,一个递增子序列ve1,一个递减子序列ve0。现在有一个数a[i],放入规则如下
- 如果a[i]大于ve1中最后一个数,且大于ve0中最后一个数,那么只能将a[i]放入ve1中
- 如果a[i]小于ve0中最后一个数,且小于ve1中最后一个数,那么只能将a[i]放入ve0中
- 如果a[i]小于ve0中最后一个数,且大于ve1中最后一个数,则比较第i+1个数,如果a[i+1]大于a[i],则将a[i]放入ve1中否则放入ve0中
那么第三条规则是怎么得到的呢,首先假设a[i]小于a[i+1],如果a[i+1]小于ve0最后一个数,那么可以将a[i+1]放入递减数列中,这时a[i]肯定不能在a[i+1]之前就放入递减数列,a[i]只能放入递增数列中。如果当a[i]大于ve1最后一个数字那么a[i+1]也大于ve1最后一个数字,那么可以将a[i+1]放入ve1中,a[i]在a[i]+1之前就放入递增数列也不会影响。综上所述只要a[i]小于a[i+1],并且a[i]在两个数列都可以放入的时候将a[i]放入递增数列。
#include <bits/stdc++.h>
using namespace std;
const int maxn = 2e5+100;
int num[maxn], n, ans[maxn];//num记录原始数字,ans记录每个数字的归属
vector <int> ve[2];//ve[0]维护递减子序列,ve[1]维护递增子序列
void init() {
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%d", &num[i]);
ve[0].push_back(1e9);
ve[1].push_back(-1);
}
int solve() {
for(int i=1;i<=n;i++) {
if((num[i] > *ve[1].rbegin() && num[i] < *ve[0].rbegin())) {
if(num[i+1] > num[i]) {
ve[1].push_back(num[i]);
ans[i] = 0;
} else {
ve[0].push_back(num[i]);
ans[i] = 1;
}
} else if(num[i] < *ve[0].rbegin()) {
ans[i] = 1;
ve[0].push_back(num[i]);
} else if(num[i] > *ve[1].rbegin()) {
ans[i] = 0;
ve[1].push_back(num[i]);
} else {
return 0;
}
}
return 1;
}
void Print() {
puts("YES");
for(int i=1;i<=n;i++) {
printf("%d ", ans[i]);
}
}
int main() {
// freopen("1.in", "r", stdin);
init();
int flag = solve();
if(!flag) {
puts("NO");
return 0;
}
Print();
return 0;
}