Bzoj 3523 [Poi2014]Bricks

最新推荐文章于 2018-10-07 22:08:00 发布

原创最新推荐文章于 2018-10-07 22:08:00 发布 · 621 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#贪心 #堆 #poi

贪心同时被 2 个专栏收录

9 篇文章

订阅专栏

堆

4 篇文章

订阅专栏

本文介绍了一个解决颜色匹配问题的贪心算法实现，通过维护一个最大堆来选择与前一个元素颜色不同的最多数量的颜色。算法考虑了尾部颜色的优先级，并解决了当堆顶元素与前一元素颜色相同时的特殊情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

原题网址：http://www.lydsy.com/JudgeOnline/problem.php?id=3523
很显然的贪心，每次找不和前面元素重复的最多的，尾部颜色在数量相同时优先级更高。WA了好多发因为姿势丑，如果堆顶元素和前面元素一样只要先把堆顶拿出来就好了。。

type
  rec=record
        c,cnt:longint;
      end;  
const
  MAXN=1000050;
var
  heap:array[0..MAXN] of rec;
  a,c,cp:array[0..MAXN] of longint;
  n,m,st,nd,i,cnt:longint;
  t:rec;
procedure swp(var a,b:longint);
  var t:longint;
  begin t:=a;a:=b;b:=t; end;
procedure swap(a,b:longint);
  begin
    swp(heap[a].c,heap[b].c);
    swp(heap[a].cnt,heap[b].cnt);
  end;
operator >(a,b:rec)c:boolean;
  begin exit((a.cnt>b.cnt)or(a.cnt=b.cnt)and(a.c=nd));end;  
procedure heapup(x:longint);
  begin
    while ((x>1)and(heap[x]>heap[x>>1])) do
      begin
        swap(x,x>>1);
        x:=x>>1;
      end;
  end;
procedure heapdown(x:longint);
  var
    e:longint;
  begin
    while ((x<<1<=cnt)and(heap[x<<1]>heap[x])or(x<<1+1<=cnt)and(heap[x<<1+1]>heap[x])) do
      begin
        e:=x<<1+ord((x<<1+1<=cnt)and(heap[x<<1+1].cnt>heap[x<<1].cnt));
        swap(x,e);
        x:=e;
      end;
  end;
procedure push(x:rec);
  begin
    inc(cnt);
    heap[cnt]:=x;
    heapup(cnt);
  end;
procedure poop;
  begin
    heap[1]:=heap[cnt];
    dec(cnt);
    heapdown(1);
  end;
procedure no_solution;
  begin
    writeln(0);
    halt;
  end;
begin
  read(n,st,nd);m:=0;
  for i:=1 to n do read(a[i]);
  cp:=a;
  for i:=1 to n do inc(m,a[i]);
  dec(a[st]);if (a[st]<0) then no_solution;
  dec(a[nd]);if (a[nd]<0) then no_solution;
  for i:=1 to n do
    if (a[i]>0) then
      begin
        t.c:=i;
        t.cnt:=a[i];
        push(t);
      end;
  c[1]:=st;c[m]:=nd;
  for i:=2 to m-1 do
    if (heap[1].c<>c[i-1])
      then
        begin
          c[i]:=heap[1].c;
          dec(heap[1].cnt);
          if (heap[1].cnt=0)
            then poop
            else heapdown(1);
        end
      else
        begin
          if (cnt=1) then no_solution;
          t:=heap[1];poop;
          c[i]:=heap[1].c;
          dec(heap[1].cnt);
          if (heap[1].cnt=0)
            then poop
            else heapdown(1);
          push(t);  
        end;
  if (c[m]=c[m-1]) then no_solution;  
  for i:=1 to m-1 do
    write(c[i],' ');
  writeln(c[m]);
end.