結果

問題 No.5007 Steiner Space Travel
ユーザー allegrogikenallegrogiken
提出日時 2022-07-30 17:58:50
言語 D
(dmd 2.106.1)
結果
AC  
実行時間 886 ms / 1,000 ms
コード長 7,215 bytes
コンパイル時間 4,603 ms
実行使用メモリ 6,952 KB
スコア 7,366,097
最終ジャッジ日時 2023-06-01 00:00:00
合計ジャッジ時間 31,945 ms
ジャッジサーバーID
(参考情報)
judge9 / judge10
純コード判定しない問題か言語
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 856 ms
4,904 KB
testcase_01 AC 829 ms
6,948 KB
testcase_02 AC 860 ms
6,948 KB
testcase_03 AC 823 ms
6,948 KB
testcase_04 AC 884 ms
4,900 KB
testcase_05 AC 828 ms
4,908 KB
testcase_06 AC 854 ms
4,908 KB
testcase_07 AC 838 ms
4,900 KB
testcase_08 AC 861 ms
4,900 KB
testcase_09 AC 841 ms
4,904 KB
testcase_10 AC 827 ms
4,904 KB
testcase_11 AC 857 ms
4,900 KB
testcase_12 AC 827 ms
4,904 KB
testcase_13 AC 855 ms
4,904 KB
testcase_14 AC 825 ms
6,948 KB
testcase_15 AC 886 ms
4,908 KB
testcase_16 AC 821 ms
4,904 KB
testcase_17 AC 857 ms
4,900 KB
testcase_18 AC 827 ms
6,548 KB
testcase_19 AC 853 ms
4,900 KB
testcase_20 AC 823 ms
4,908 KB
testcase_21 AC 854 ms
6,952 KB
testcase_22 AC 852 ms
6,952 KB
testcase_23 AC 855 ms
6,948 KB
testcase_24 AC 828 ms
6,948 KB
testcase_25 AC 830 ms
6,948 KB
testcase_26 AC 823 ms
6,952 KB
testcase_27 AC 827 ms
6,952 KB
testcase_28 AC 830 ms
4,908 KB
testcase_29 AC 826 ms
4,900 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
/dmd2/linux/bin64/../../src/phobos/std/numeric.d(2967): Warning: cannot inline function `std.numeric.gcdImpl!ulong.gcdImpl`

ソースコード

diff #

void main() { runSolver(); }

// ----------------------------------------------

enum long ALPHA = 5;
enum ALPHAS = [0, 0, ALPHA^^2, ALPHA, 1];

alias Vector = Vector2!long;

struct Spot {
  int type, id;
  Vector location;

  long energy(Spot other) {
    auto k = ALPHAS[type + other.type];
    return location.norm(other.location) * k;
  }

  string toString() {
    return "%s %s".format(type, id);
  }
}
enum long BOUNDARY = 10 ^^ 3 + 1;

Vector center(Vector[] arr) {
  long x = arr.map!"a.x".sum / arr.length;
  long y = arr.map!"a.y".sum / arr.length;
  return Vector(x, y);
}

Vector center(Spot[] arr) {
  long x = arr.map!"a.location.x".sum / arr.length;
  long y = arr.map!"a.location.y".sum / arr.length;
  return Vector(x, y);
}

auto degComp(Spot as, Spot bs, Vector center) {
  auto a = as.location;
  auto b = bs.location;
  auto d1 = atan2(a.y.to!real - center.y.to!real, a.x.to!real - center.x.to!real);
  auto d2 = atan2(b.y.to!real - center.y.to!real, b.x.to!real - center.x.to!real);
  return d1 < d2;
}

void problem() {
  auto N = scan!int;
  auto M = scan!int;
  auto P = N.iota.map!(_ => Vector(scan!long, scan!long)).array;
  auto rnd = Xorshift(unpredictableSeed);

  auto solve() {
    auto allCenter = center(P);
    auto targets = P.enumerate(1).map!(p => Spot(1, p[0], p[1])).array;

    targets.sort!((a, b) => degComp(a, b, allCenter));
    auto startIndex = targets.countUntil!(t => t.id == 1);
    targets = targets[startIndex..$] ~ targets[0..startIndex];

    Vector[] stationLocations = [targets[0].location];
    auto rest = targets[1..$];
    foreach(m; 1..M) {
      long best = long.max;
      long bestX, bestY;
      foreach(x; iota(0, BOUNDARY, 10)) foreach(y; iota(0, BOUNDARY, 10)) {
        auto base = Vector(x, y);
        auto heap = (new long[](0)).heapify!"a > b";
        foreach(r; rest) {
          heap.insert(base.norm(r.location));
        }
        auto border = heap.take(10).array[$ - 1];
        if (best.chmin(border)) {
          bestX = x;
          bestY = y;
        }
      }

      auto base = Vector(bestX, bestY);
      rest = rest.filter!(r => base.norm(r.location) > best).array;
      stationLocations ~= base;
    }

    auto stationIndicies = 0 ~ iota(1, M + 1).map!(m => N * m / M).array;
    auto stations = stationLocations.enumerate(1).map!(s => Spot(2, s[0], s[1])).array;

    Spot[] routes = targets.dup ~ targets[0];
    auto ri = iota(1, routes.length - 1).array;
    auto rs = M.iota.array;
    foreach(i; 0..10^^6 / 3) {
      auto a = ri.choice(rnd);
      auto b = ri.choice(rnd);

      auto efa = routes[a - 1].energy(routes[a]) + routes[a].energy(routes[a + 1]);
      auto efb = routes[b - 1].energy(routes[b]) + routes[b].energy(routes[b + 1]);
      auto eta = routes[a - 1].energy(routes[b]) + routes[b].energy(routes[a + 1]);
      auto etb = routes[b - 1].energy(routes[a]) + routes[a].energy(routes[b + 1]);

      if (efa + efb > eta + etb) swap(routes[a], routes[b]);
    }
    ri = iota(1, routes.length).array;
    foreach(i; 0..10^^5) {
      auto a = ri.choice(rnd);
      auto s = rs.choice(rnd);

      auto ef = routes[a - 1].energy(routes[a]);
      auto et = routes[a - 1].energy(stations[s]) + stations[s].energy(routes[a]);

      if (ef > et) {
        routes = routes[0..a] ~ stations[s] ~ routes[a..$];
        ri ~= ri.length;
      }
    }
    ri = iota(1, routes.length - 1).array;
    foreach(i; 0..10^^6 / 3) {
      auto a = ri.choice(rnd);
      auto b = ri.choice(rnd);

      auto efa = routes[a - 1].energy(routes[a]) + routes[a].energy(routes[a + 1]);
      auto efb = routes[b - 1].energy(routes[b]) + routes[b].energy(routes[b + 1]);
      auto eta = routes[a - 1].energy(routes[b]) + routes[b].energy(routes[a + 1]);
      auto etb = routes[b - 1].energy(routes[a]) + routes[a].energy(routes[b + 1]);

      if (efa + efb > eta + etb) swap(routes[a], routes[b]);
    }

    Spot[] filtered;
    filtered ~= routes[0];
    routes = routes.uniq.array;
    foreach(i; 1..routes.length - 1) {
      if (routes[i - 1..i + 1].all!(s => s.type == 2)) {
        if (routes[i - 1].id == routes[i + 1].id) continue;
      }

      filtered ~= routes[i];
    }
    filtered ~= routes[$ - 1];
    routes = filtered;

    long energy;
    foreach(i; 0..routes.length - 1) {
      energy += routes[i].energy(routes[i + 1]);
    }
    stderr.writeln(10L^^9 / (1000 + energy.to!real.sqrt));

    string[] ans;
    foreach(s; stations) ans ~= s.location.toString;
    ans ~= routes.length.to!string;
    ans ~= routes.map!"a.toString".array;

    debug {} else return ans;
  }

  outputForAtCoder(&solve);
}

// ----------------------------------------------

import std.stdio, std.conv, std.array, std.string, std.algorithm, std.container, std.range, std.math, std.typecons, std.numeric, std.traits, std.functional, std.bigint, std.datetime.stopwatch, core.time, core.bitop, std.random;
string scan(){ static string[] ss; while(!ss.length) ss = readln.chomp.split; string res = ss[0]; ss.popFront; return res; }
T scan(T)(){ return scan.to!T; }
T[] scan(T)(long n){ return n.iota.map!(i => scan!T()).array; }
void deb(T ...)(T t){ debug { write("#"); writeln(t); }}
T[] divisors(T)(T n) { T[] ret; for (T i = 1; i * i <= n; i++) { if (n % i == 0) { ret ~= i; if (i * i != n) ret ~= n / i; } } return ret.sort.array; }
bool chmin(T)(ref T a, T b) { if (b < a) { a = b; return true; } else return false; }
bool chmax(T)(ref T a, T b) { if (b > a) { a = b; return true; } else return false; }
string charSort(alias S = "a < b")(string s) { return (cast(char[])((cast(byte[])s).sort!S.array)).to!string; }
ulong comb(ulong a, ulong b) { if (b == 0) {return 1;}else{return comb(a - 1, b - 1) * a / b;}}
string toAnswerString(R)(R r) { return r.map!"a.to!string".joiner(" ").array.to!string; }
void outputForAtCoder(T)(T delegate() fn) {
  static if (is(T == float) || is(T == double) || is(T == real)) "%.16f".writefln(fn());
  else static if (is(T == void)) fn();
  else static if (is(T == string)) fn().writeln;
  else static if (isInputRange!T) {
    static if (!is(string == ElementType!T) && isInputRange!(ElementType!T)) foreach(r; fn()) r.toAnswerString.writeln;
    else foreach(r; fn()) r.writeln;
  }
  else fn().writeln;
}
void runSolver() {
  enum BORDER = "#==================================";
  debug { BORDER.writeln; while(true) { "#<<< Process time: %s >>>".writefln(benchmark!problem(1)); BORDER.writeln; } }
  else problem();
}
enum YESNO = [true: "Yes", false: "No"];

// -----------------------------------------------

struct Vector2(T) {
  T x, y;
  Vector2 add(Vector2 other) { return Vector2(x + other.x, y + other.y ); }
  Vector2 opAdd(Vector2 other) { return add(other); }
  Vector2 sub(Vector2 other) { return Vector2(x - other.x, y - other.y ); }
  Vector2 opSub(Vector2 other) { return sub(other); }
  T norm(Vector2 other) {return (x - other.x)*(x - other.x) + (y - other.y)*(y - other.y); }
  T dot(Vector2 other) {return x*other.y - y*other.x; }
  Vector2 normalize() {if (x == 0 || y == 0) return Vector2(x == 0 ? 0 : x/x.abs, y == 0 ? 0 : y/y.abs);const gcd = x.abs.gcd(y.abs);return Vector2(x / gcd, y / gcd);}
  string toString() { return "%s %s".format(x, y); }
}

0