【BZOJ】3523: [Poi2014]Bricks

题意

\(n(n \le 1000000)\)个物品，颜色分别为\(a[i]\)，现在要求排在一排使得相邻两个砖块的颜色不同，且限定第一个砖块和最后一个砖块的颜色，输出一个合法解否则输出-1。

分析

贪心？反正我不会证明。

题解

从左到右扫，每次放一个数量最多且合法的砖块，如果数量相同，优先放末尾的。

#include <bits/stdc++.h>
using namespace std;
char Buf[15000000], *buf=Buf;
int getint() {
int x=0;
char c=*buf++;
for(; c<'0'||c>'9'; c=*buf++);
for(; c>='0'&&c<='9'; x=x*10+c-'0', c=*buf++);
return x;
}
typedef pair<int, int> pii;
#define mkpii(a, b) make_pair<int, int> (a, b)
const int N=1000005;
int c
, n, st, ed, m, g=1;
priority_queue<pii> q;
int main() {
fread(Buf, 1, sizeof Buf, stdin);
n=getint(), st=getint(), ed=getint();
for(int i=1; i<=n; ++i) {
int x=getint();
m+=x;
x-=(int)(i==st)+(i==ed);
if(x>0) {
q.push(mkpii(x, i==ed?n+i:i));
}
if(x<0) {
g=0;
}
}
if(m==1) {
g=1;
}
c[1]=st==ed?n+st:st;
c[m]=n+ed;
for(int i=2; i<m; ++i) {
pii t=q.top();
q.pop();
if(c[i-1]==t.second && q.size()) {
pii t2=q.top();
q.pop();
q.push(t);
t=t2;
}
c[i]=t.second;
if(t.first-1) {
q.push(mkpii(t.first-1, t.second));
}
}
for(int i=1; i<m; ++i) {
if(c[i]==c[i+1]) {
g=0;
break;
}
}
if(!g) {
puts("0");
return 0;
}
for(int i=1; i<=m; ++i) {
int id=c[i];
printf("%d ", id>n?id-n:id);
}
return 0;
}