CodeForces 140C New Year Snowmen（堆）

Posted 2021-02-07 water-mi

题面

CodeForces

题解

因为要保证两两不同，所以不能单纯的开堆来维护，堆维护一个二元组，个数为第一关键字，编号为第二关键字，对于一个相同的颜色，统计一下这个颜色的个数再用堆来维护就好了。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using std::min; using std::max;
using std::sort; using std::swap;
using std::unique; using std::lower_bound;
using std::priority_queue;
typedef long long ll;

template<typename T>
void read(T &x) {
    int flag = 1; x = 0; char ch = getchar();
    while(ch < '0' || ch > '9') { if(ch == '-') flag = -flag; ch = getchar(); }
    while(ch >= '0' && ch <= '9') x = x * 10 + ch - '0', ch = getchar(); x *= flag;
}

const int N = 1e5 + 10;
int n, a[N], b[N], zc[N], col[N];
struct Memb { int val, ind; };
struct Ans { int x, y, z; } A[N]; int tot;
inline bool operator < (const Memb &a, const Memb &b) { return a.val < b.val; }
priority_queue<Memb> q;

int main () {
    read(n); int m = n;
    for(int i = 1; i <= n; ++i)
        read(a[i]), b[i] = a[i];
    sort(&b[1], &b[m + 1]), m = unique(&b[1], &b[m + 1]) - b - 1;
    for(int i = 1; i <= n; ++i) {
        int tmp = lower_bound(&b[1], &b[m + 1], a[i]) - b;
        zc[tmp] = a[i], a[i] = tmp, ++col[tmp];
    }
    for(int i = 1; i <= m; ++i)
        q.push((Memb){col[i], i});
    Memb a_, b_, c_;
    while(q.size() >= 3) {
        a_ = q.top(); q.pop();
        b_ = q.top(); q.pop();
        c_ = q.top(); q.pop();
        A[++tot] = (Ans){zc[a_.ind], zc[b_.ind], zc[c_.ind]};
        if(--a_.val) q.push(a_);
        if(--b_.val) q.push(b_);
        if(--c_.val) q.push(c_);
    }
    printf("%d
", tot);
    for(int i = 1; i <= tot; ++i) {
        if(A[i].y > A[i].x) swap(A[i].y, A[i].x);
        if(A[i].z > A[i].y) swap(A[i].z, A[i].y);
        if(A[i].y > A[i].x) swap(A[i].y, A[i].x);
        if(A[i].z > A[i].y) swap(A[i].z, A[i].y);
        printf("%d %d %d
", A[i].x, A[i].y, A[i].z);
    }
    return 0;
}