結果

問題 No.168 ものさし
ユーザー t8m8⛄️
提出日時 2016-11-27 22:47:46
言語 Nim
(2.2.0)
結果
AC  
実行時間 187 ms / 2,000 ms
コード長 1,582 bytes
コンパイル時間 4,282 ms
コンパイル使用メモリ 66,612 KB
実行使用メモリ 27,580 KB
最終ジャッジ日時 2024-06-29 20:38:00
合計ジャッジ時間 6,498 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

import strutils, sequtils, algorithm, math

type
  DisjointSet = ref object
    par: seq[int]

proc newDisjointSet*(n: int): DisjointSet =
  result = DisjointSet(par: newSeq[int](n))
  for i in 0..n-1:
    result.par[i] = i

proc find*(self: DisjointSet, x: int): int =
  if self.par[x] == x:
    result = x
  else:
    self.par[x] = self.find(self.par[x])
    result = self.par[x]

proc same*(self: DisjointSet, x, y: int): bool =
  result = self.find(x) == self.find(y)

proc unite*(self: DisjointSet, x, y: int) =
  let
    s = self.find(x)
    t = self.find(y)
  if s != t: self.par[s] = t

proc `$`*(self: DisjointSet): string =
  result = $self.par

proc kruskal*(n: int, s, t, c: seq[int]): int64 =
  var
    m = s.len
    edges = newSeq[int64](m)

  for i in 0..m-1:
    edges[i] = c[i].int64 shl 32 or i
  edges.sort(cmp[int64])

  var ds = newDisjointSet(n)

  for i in 0..m-1:
    var cur = cast[int32](edges[i])
    if not ds.same(s[cur], t[cur]):
      ds.unite(s[cur], t[cur])
      result = c[cur].int64
    if ds.same(0, n-1): break

when isMainModule:
  var
    n = stdin.readLine.parseInt
    p = newSeqWith(n, newSeq[int](2))

  for i in 0..n-1:
    p[i] = stdin.readLine.split.map(parseInt)

  var
    s: seq[int] = @[]
    t: seq[int] = @[]
    c: seq[int] = @[]
  for i in 0..n-1:
    for j in i+1..n-1:
      s.add(i)
      t.add(j)
      var
        dx = p[i][0] - p[j][0]
        dy = p[i][1] - p[j][1]
        dd = dx*dx + dy*dy
        d = sqrt(dd.float).int64
      while dd > d*d: d += 1
      c.add(int((d+9) div 10 * 10))


  echo kruskal(n, s, t, c)
0