2018CCPC吉林赛区（重现赛）C. TJustice (贪心)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=6557

解题心得：

• 做法就是将输入的值按照升序排序，然后从小到大依次合并，例如$3$和$3$合并成$2$，$4$和$4$合并成$3$，用一个栈来模拟一下。如果能合并出两个$1$那么输出$YES$否则输出$NO$，如果只有一个$1$那么无论栈中还剩下什么求和之后都一定小于二分之一，这里可以等比数列求和再求个极限就可以证明出来。在记录的时候合并起来变成$1$的肯定是排序后连续的一段区间，只需要记录一下这个段区间就好。有一个小细节，只能从小到大合并，如果从大到小可能会漏掉这也很容易想到。
• 在比赛的时候明明都有思路了开始敲了，队友一直和我说着说那，然后又怀疑我的思路，最后给我提供了个超级复杂的路径记录的方法，然后就写炸了。最后没办法写不下去了，删了代码，不理他BB，十多分钟就敲出来了，难受。

---

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e6+100;

struct Node {
    int va, pos;//记录输入的值和原本所在的位置

    bool operator < (const Node &x) const {
        if(va != x.va) return va > x.va;
        return pos < x.pos;
    }
}node[maxn];

int n;
bool vis[maxn];

void init() {
    scanf("%d", &n);
    for(int i=1;i<=n;i++) {
        scanf("%d", &node[i].va);
        node[i].pos = i;
    }
    sort(node+1, node+1+n);
}

stack <pair<int,int> > st;
vector <pair<int,int> > ans;

void solve() {
    ans.clear();
    while(!st.empty()) st.pop();
    for(int i=1;i<=n;i++) {
        if(node[i].va == 1) {
            ans.push_back(make_pair(i, i));
            continue;
        }
        if(ans.size() >= 2) break;
        if(st.empty()) {
            st.push(make_pair(node[i].va, i));
        } else {
            int va = node[i].va, pos = i;
            while(!st.empty() && va == st.top().first && va != 1) {
                va--;
                pos = st.top().second;
                st.pop();
            }

            if(va == 1) {
                ans.push_back(make_pair(pos , i));
            } else {
                st.push(make_pair(va, pos));
            }
            if(ans.size() == 2) break;
        }
    }

    if(ans.size() < 2) {
        printf("NO\n");
        return ;
    }

    printf("YES\n");
    for(int i=ans[0].first;i<=ans[0].second;i++) {
        vis[node[i].pos] = true;
    }
    for(int i=1;i<=n;i++) {
        if(vis[i]) {
            printf("1");
            vis[i] = false;
        } else {
            printf("0");
        }
    }
    printf("\n");

}

int main() {
    //freopen("1.in.txt", "r", stdin);
    int t; scanf("%d", &t);
    int T = t;
    while(t--) {
        init();
        printf("Case %d: ", T-t);
        solve();
    }
    return 0;
}