結果

問題 No.261 ぐるぐるぐるぐる！あみだくじ！
コンテスト
ユーザー te-sh
提出日時 2017-06-02 15:26:58
言語 D
(dmd 2.109.1)
結果
AC  
実行時間 6 ms / 5,000 ms
コード長 1,727 bytes
コンパイル時間 1,670 ms
コンパイル使用メモリ 173,640 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-06-12 19:42:05
合計ジャッジ時間 2,973 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std.algorithm, std.conv, std.range, std.stdio, std.string;
import std.typecons;  // Tuple, Nullable, BigFlags
import std.bigint;    // BigInt

void main()
{
  auto n = readln.chomp.to!size_t;
  auto k = readln.chomp.to!size_t;
  auto bi = k.iota.map!(_ => readln.split.to!(size_t[]).front - 1).array;

  auto ci = n.iota.array;
  foreach (b; bi)
    swap(ci[b], ci[b+1]);

  auto di = n.iota.map!(i => ci.countUntil(i));

  auto ei = new size_t[][](n);
  foreach (i, ref e; ei) {
    for (auto j = i; e.empty || j != i; j = di[j])
      e ~= j;
  }

  auto yi = ei.map!(e => e.length.to!BigInt).array;

  auto q = readln.chomp.to!size_t;
  foreach (_; 0..q) {
    auto ai = readln.split.to!(size_t[]).map!"a - 1".array;
    auto fi = n.iota.map!(i => ai.countUntil(i));
    auto xi = zip(ei, fi).map!(ef => ef[0].countUntil(ef[1]).to!BigInt).array;
    if (xi.any!"a == -1")
      writeln(-1);
    else
      writeln(calc1(xi, yi));
  }
}

auto calc1(T)(T[] xi, T[] yi)
{
  auto z = xi[0], l = yi[0];

  foreach (i; 1..xi.length) {
    auto r = calc2(z, l, xi[i], yi[i]); z = r[0]; l = r[1];
    if (l == -1)
      return T(-1);
  }

  return z == 0 ? l : z;
}

auto calc2(T)(T x1, T y1, T x2, T y2)
{
  auto g = euclid(y1, y2);
  if (x1 % g != x2 % g) return Tuple!(T, T)(T(-1), T(-1));

  T m, n;
  exEuclid(y1, y2, m, n);
  m *= (x2 - x1) / g;

  auto l = y1 / g * y2;

  return Tuple!(T, T)(((m * y1 + x1) % l + l) % l, l);
}

pure T euclid(T)(T a, T b)
{
  if (a < b) return euclid(b, a);
  auto c = a % b;
  return c == 0 ? b : euclid(b, c);
}

pure T exEuclid(T)(T a, T b, ref T x, ref T y)
{
  auto g = a;
  x = 1;
  y = 0;
  if (b != 0) {
    g = exEuclid(b, a % b, y, x);
    y -= a / b * x;
  }
  return g;
}
0