結果

問題 No.168 ものさし
ユーザー te-sh
提出日時 2017-02-06 18:53:23
言語 D
(dmd 2.109.1)
結果
AC  
実行時間 126 ms / 2,000 ms
コード長 1,667 bytes
コンパイル時間 850 ms
コンパイル使用メモリ 116,364 KB
実行使用メモリ 15,520 KB
最終ジャッジ日時 2024-06-12 06:55:51
合計ジャッジ時間 2,483 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std.algorithm, std.conv, std.range, std.stdio, std.string;
import std.math;      // math functions

alias Point!long point;

void main()
{
  auto n = readln.chomp.to!size_t;
  auto pi = n.iota.map!(_ => readln.split.to!(int[])).map!(rd => point(rd[0], rd[1])).array;

  auto pli = new pointLen[](n * (n - 1) / 2);
  auto k = size_t(0);
  foreach (i; 0..n-1)
    foreach (j; i+1..n)
      pli[k++] = pointLen(i, j, (pi[i] - pi[j]).hypot2);

  pli.sort!"a.len2 < b.len2";

  auto calc() {
    auto uf = new UnionFind!size_t(n);
    foreach (pl; pli) {
      uf.unite(pl.i, pl.j);
      if (uf.find(0) == uf.find(n - 1)) return pl.len2;
    }
    return 0L;
  }

  auto sq(long len2, long a) {
    return a * a >= len2;
  }

  auto len2 = calc;
  auto r = iota(10L, pli.back.len2.to!real.sqrt.to!long + 20, 10L).assumeSorted!sq.upperBound(len2);

  writeln(r.front);
}

struct pointLen {
  size_t i, j;
  long len2;
}

struct Point(T) {
  T x, y;

  auto opBinary(string op: "+")(Point!T rhs) { return Point!T(x + rhs.x, y + rhs.y); }
  auto opBinary(string op: "-")(Point!T rhs) { return Point!T(x - rhs.x, y - rhs.y); }
  auto opBinary(string op: "*")(Point!T rhs) { return x * rhs.x + y * rhs.y; }
  auto opBinary(string op: "*")(T a) { return Point!T(x * a, y * a); }

  T hypot2() { return x ^^ 2 + y ^^ 2; }
}

class UnionFind(T) {
  T[] p; // parent
  T s; // sentinel

  this(T n) {
    p = new T[](n);
    s = n + 1;
    p[] = s;
  }

  T find(T i) {
    if (p[i] == s) {
      return i;
    } else {
      p[i] = find(p[i]);
      return p[i];
    }
  }

  void unite(T i, T j) {
    auto pi = find(i), pj = find(j);
    if (pi != pj) p[pj] = pi;
  }
}
0