[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