結果

問題 No.366 ロボットソート
ユーザー te-sh
提出日時 2017-07-18 16:53:16
言語 D
(dmd 2.109.1)
結果
AC  
実行時間 3 ms / 2,000 ms
コード長 643 bytes
コンパイル時間 817 ms
コンパイル使用メモリ 107,540 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-06-12 20:59:55
合計ジャッジ時間 1,983 ms
ジャッジサーバーID
(参考情報)
judge4 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff #

import std.algorithm, std.conv, std.range, std.stdio, std.string;

void main()
{
  auto rd = readln.split.to!(int[]), n = rd[0], k = rd[1];
  auto a = readln.split.to!(int[]);

  auto idx = 0;
  int[int] buf;
  auto c = a.dup; c.sort();
  foreach (ci; c) buf[ci] = idx++;
  foreach (ref ai; a) ai = buf[ai];

  auto b = new int[][](n);
  foreach (i, ai; a) {
    if (ai % k != i % k) {
      writeln(-1);
      return;
    }
    b[i % k] ~= ai / k;
  }
  writeln(b.map!(bi => calc(bi)).sum);
}

auto calc(int[] a)
{
  auto n = a.length, r = 0;
  foreach (i; 0..n) {
    auto j = a.countUntil(i);
    r += j;
    a.remove(j);
  }
  return r;
}
0