【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=x10+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], n, st, ed, m, g=1;
priority_queue 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)(ist)+(ied);
if(x>0) {
q.push(mkpii(x, ied?n+i:i));
}
if(x<0) {
g=0;
}
}
if(m1) {
g=1;
}
c[1]=sted?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;
}