結果

問題 No.168 ものさし
コンテスト
ユーザー t8m8⛄️
提出日時 2017-08-19 20:50:32
言語 Nim
(2.2.0)
結果
RE  
実行時間 -
コード長 1,580 bytes
コンパイル時間 3,355 ms
コンパイル使用メモリ 66,688 KB
実行使用メモリ 43,264 KB
最終ジャッジ日時 2024-06-30 02:56:43
合計ジャッジ時間 5,882 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample RE * 4
other WA * 3 RE * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

import strutils, sequtils, algorithm, math

type
  DisjointSet = ref object of RootObj
    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*[T](n: int, s, t: seq[int], c: seq[T]): T =
  var edges = newSeq[(int, int)](s.len)

  for i in 0..<s.len:
    edges[i] = (c[i], i)
  edges = edges.sorted(cmp)

  var ds = newDisjointSet(n)

  for i in 0..<s.len:
    var cur = cast[int32](edges[i])
    if not ds.same(s[cur], t[cur]):
      ds.unite(s[cur], t[cur])
      result = c[cur]
    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