BZOJ3523: [Poi2014]Bricks

#include<iostream>
#include<cstdio>
#include<queue>
#define N 1000010
using namespace std;
struct ppp{int w,c;};
int st,en;
bool operator <(ppp x,ppp y)
{
if(x.w!=y.w) return x.w<y.w;
if(x.c==en) return false;
return true;
}
priority_queue<ppp>q;
int ans
;
int main()
{
int n;
scanf("%d%d%d",&n,&st,&en);
int i,j,x,y,m=0;
ppp t,tmp;
for(i=1;i<=n;i++)
{
scanf("%d",&t.w);
m+=t.w;
t.c=i;
if(i==st) t.w--;
if(i==en) t.w--;
if(t.w<0) {puts("0");return 0;}
q.push(t);
}
ans[1]=st;ans[m]=en;
bool f;
for(i=2;i<m;i++)
{
t=q.top();f=false;
q.pop();
if(t.c==st)
{
tmp=t;
if(!q.empty()) t=q.top();
else {puts("0");return 0;}
q.pop();
f=true;
}
st=ans[i]=t.c;
if(t.w>1)
q.push((ppp){t.w-1,t.c});
if(f) q.push(tmp);
}
if(ans[m-1]==ans[m]) {puts("0");return 0;}
for(i=1;i<=m;i++)
printf("%d ",ans[i]);

}