結果

問題 No.318 学学学学学
ユーザー t8m8⛄️
提出日時 2017-06-09 15:57:32
言語 Nim
(2.2.0)
結果
WA  
実行時間 -
コード長 2,497 bytes
コンパイル時間 3,932 ms
コンパイル使用メモリ 73,472 KB
実行使用メモリ 23,412 KB
最終ジャッジ日時 2024-06-30 01:18:14
合計ジャッジ時間 7,486 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 22 WA * 4
権限があれば一括ダウンロードができます
コンパイルメッセージ
/home/judge/data/code/Main.nim(1, 39) Warning: Use the new 'sugar' module instead; future is deprecated [Deprecated]
/home/judge/data/code/Main.nim(94, 12) Warning: Deprecated since v0.20, use 'initHashSet'; initSet is deprecated [Deprecated]

ソースコード

diff # 

import strutils, sequtils, algorithm, future, tables, sets

type
  PrirorityQueue*[T] = ref object of RootObj
    elms: seq[T]
    cmp: (T, T) -> int

proc newPriorityQueue*[T](cmp: (T, T) -> int): PrirorityQueue[T] =
  new(result)
  result.elms = newSeq[T]()
  result.cmp = cmp

template parent(cur: int): int =
  (cur - 1) div 2

template childl(cur: int): int =
  cur * 2 + 1

template childr(cur: int): int =
  cur * 2 + 2

proc upHeap[T](self: var PrirorityQueue[T], cur: var int) =
  var par = parent(cur)

  while par >= 0 and self.cmp(self.elms[cur], self.elms[par]) < 0:
    swap(self.elms[cur], self.elms[par])
    (cur, par) = (par, parent(par))

proc downHeap[T](self: var PrirorityQueue[T], cur: var int) =
  var (chl, chr) = (childl(cur), childr(cur))

  while chl < self.elms.len:
    var ch =
      if chr >= self.elms.len or self.cmp(self.elms[chl], self.elms[chr]) < 0: chl
      else: chr

    if self.cmp(self.elms[cur], self.elms[ch]) < 0:
      break

    swap(self.elms[cur], self.elms[ch])
    (cur, chl, chr) = (ch, childl(cur), childr(cur))

proc len*[T](self: PrirorityQueue[T]): int =
  self.elms.len

proc peek*[T](self: var PrirorityQueue[T]): T =
  self.elms[0]

proc push*[T](self: var PrirorityQueue[T], data: T) =
  var
    n = self.elms.len
    cur = n
  self.elms.setLen(n + 1)
  self.elms[cur] = data
  self.upHeap(cur)

proc pop*[T](self: var PrirorityQueue[T]): T =
  assert self.elms.len > 0
  var
    n = self.elms.len
    cur = 0
  result = self.elms[0]
  self.elms[cur] = self.elms[n-1]
  self.elms.del(n-1)
  self.downHeap(cur)

proc delete*[T](self: var PrirorityQueue[T], data: T) =
  var
    n = self.elms.len
    idx = -1
  for i in 0..<n:
    if self.elms[i] == data:
      idx = i
      break
  if idx == -1:
    return
  elif idx == n - 1:
    self.elms.del(idx)
  else:
    self.elms[idx] = self.elms[n-1]
    self.elms.del(n-1)
    self.downHeap(idx)
    self.upHeap(idx)

# ==============================================================================

when isMainModule:
  var
    n = stdin.readLine.parseInt
    a = stdin.readLine.split.map(parseInt)
    b = newSeq[int](n)
    lastIdx = newTable[int, int]()
    pQ = newPriorityQueue[int]((x: int, y: int) => y - x)
    used = initSet[int]()

  for i in 0..<n:
    lastIdx[a[i]] = i

  for i in 0..<n:
    if not used.contains(a[i]):
      pQ.push(a[i])
      used.incl(a[i])

    b[i] = pQ.peek()

    if i == lastIdx[a[i]]:
      pQ.delete(a[i])
      used.excl(a[i])

  echo b.join(" ")
0