[BZOJ3523][[Poi2014]Bricks][贪心+优先队列]

[BZOJ3523][[Poi2014]Bricks][贪心+优先队列]

题目大意：

给你每种颜色的砖块数量，相同颜色的砖块不能放在一起，两头颜色已经确定，构造一种方案

思路：

直接贪心，每次选择砖块数量最多的颜色，如果颜色相同优先考虑最后一块砖块的颜色（显然为了防止和最后一块颜色相同，要把这些砖块提前放掉。）

维护数量最多的颜色可以用优先队列，每次取出队首，如果颜色和前一块砖块颜色一样就再取队首，然后把第一次取出来的放回去。

坑

第一块和最后一块颜色可能相同，减要分别减一次再扔到队列里……

代码：

WA和RE了好多次……

#include <cstdio>
#include <queue>
using namespace std;
const int Maxn = 1000010;
inline char get(void) {
    static char buf[1000000], *p1 = buf, *p2 = buf;
    if (p1 == p2) {
        p2 = (p1 = buf) + fread(buf, 1, 1000000, stdin);
        if (p1 == p2) return EOF;
    }
    return *p1++;
}
inline void read(int &x) {
    x = 0; static char c; bool minus = false;
    for (; !(c >= '0' && c <= '9'); c = get()) if (c == '-') minus = true;
    for (; c >= '0' && c <= '9'; x = x * 10 + c - '0', c = get()); if (minus) x = -x;
}
int s, e, n, m, ans[Maxn];
struct Abcd {
    int c, n;
    friend bool operator < (const Abcd &a, const Abcd &b) {
        if (a.n == b.n) return !(a.c == e);
        else return a.n < b.n;
    }
} tmp, tmp2;
priority_queue<Abcd> qu;
int main(void) {
    //freopen("in.txt", "r", stdin);
    //freopen("out.txt", "w", stdout);
    read(n), read(s), read(e); 
    for (int i = 1; i <= n; i++) {
        read(tmp.n);
        tmp.c = i;
        if (tmp.c == s) tmp.n--;
        if (tmp.c == e) tmp.n--;
        if (tmp.n < 0) return puts("0"), 0;
        m += tmp.n;
        if (tmp.n) qu.push(tmp);
    }
    ans[0] = s, ans[m + 1] = e;
    for (int i = 1; i <= m; i++) {
        tmp = qu.top(); qu.pop();
        if (tmp.c != ans[i - 1]) {
            ans[i] = tmp.c;
            tmp.n--;
            if (tmp.n) qu.push(tmp);
        }
        else {
            if (qu.empty()) return puts("0"), 0;
            tmp2 = qu.top(); qu.pop();
            ans[i] = tmp2.c;
            tmp2.n--;
            if (tmp2.n) qu.push(tmp2);
            qu.push(tmp);
        }
    }
    if (ans[m] == ans[m + 1]) return puts("0"), 0;
    for (int i = 0; i <= m + 1; i++) printf("%d ", ans[i]);
    return 0;
}

完。

By g1n0st