結果

問題 No.9 モンスターのレベル上げ
ユーザー むらためむらため
提出日時 2017-07-31 07:01:28
言語 Nim
(2.0.2)
結果
CE  
(最新)
AC  
(最初)
実行時間 -
コード長 2,183 bytes
コンパイル時間 841 ms
コンパイル使用メモリ 68,400 KB
最終ジャッジ日時 2024-04-27 02:28:30
合計ジャッジ時間 1,161 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
コンパイルエラー時のメッセージ・ソースコードは、提出者また管理者しか表示できないようにしております。（リジャッジ後のコンパイルエラーは公開されます）
ただし、clay言語の場合は開発者のデバッグのため、公開されます。

コンパイルメッセージ
/home/judge/data/code/Main.nim(1, 50) Warning: Use the new 'sugar' module instead; future is deprecated [Deprecated]
/home/judge/data/code/Main.nim(1, 62) Error: cannot open file: queues

ソースコード

diff # 

import sequtils,strutils,strscans,algorithm,math,future,sets,queues,tables
template get():string = stdin.readLine()
template times(n:int,body:untyped): untyped = (for _ in 0..<n: body)
# TODO: proc pushpop & poppush for optimization / 蟻本の実装だとswapがいらない

type
  Heap*[T] = object
    nodes: seq[T]
    compare: proc(x,y:T):int
proc newHeap*[T](compare:proc(x,y:T):int): Heap[T] =
  Heap[T](nodes:newSeq[T](),compare:compare)
proc size*[T](h:Heap[T]):int = h.nodes.len()
proc items*[T](h:Heap[T]):seq[T] = h.nodes.sorted(h.compare) # TODO ASC?
proc peek*[T](h:Heap[T]): T = h.nodes[0]
proc push*[T](h:var Heap[T],node:T):void =
  h.nodes.add(node) #末尾に追加
  var i = h.nodes.len() - 1
  while i > 0: # 末尾から木を整形
    let parent = (i - 1) div 2
    if h.compare(h.nodes[parent],h.nodes[i]) <= 0: break
    swap(h.nodes[i],h.nodes[parent])
    i = parent
proc pop*[T](h:var Heap[T]):T =
  if h.size <= 0: raise newException(Exception,"heap is empty")
  result = h.nodes[0] # rootと末尾を入れ替えて木を整形
  h.nodes[0] = h.nodes[^1]
  h.nodes.setLen(h.nodes.len() - 1)
  let size = h.nodes.len()
  var i = 0
  while true :
    let L = i * 2 + 1
    let R = i * 2 + 2
    if L >= size : break
    let child = if R < size and h.compare(h.nodes[R],h.nodes[L]) <= 0 : R else: L
    if h.compare(h.nodes[i],h.nodes[child]) <= 0: break
    swap(h.nodes[i],h.nodes[child])
    i = child

let
  N = get().parseInt # ~1500
  A = get().split().map(parseInt) # my lv
  B = get().split().map(parseInt) #enemy lv

# 一番レベルの低い 一番戦ってないものを戦わせる
type monster = tuple[lv:int,cnt:int]
var myInitialHeap = newHeap(
  proc (a,b:monster):int =
    if a.lv == b.lv : a.cnt - b.cnt
    else: a.lv - b.lv )
for a in A: myInitialHeap.push((a,0))

var resSeq = newSeq[int]()
for i in 0..<N:
  var myHeap = myInitialHeap
  for j in 0..<N: # 1500 * 1500 * log(1500)
    let b = B[(i + j) mod N]
    var me = myHeap.pop()
    me.lv += b div 2  # (b div 2)ぶんレベルアップ
    me.cnt += 1
    myHeap.push(me)
  let count = myHeap.items().sorted((a,b)=>a.cnt - b.cnt)[^1].cnt
  resSeq.add(count)
echo resSeq.min
0