Codeforces Round #550 (Div. 3) G. Two Merged Sequences（贪心思维好题）

题目链接：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;
}