結果

問題 No.74 貯金箱の退屈
コンテスト
ユーザー te-sh
提出日時 2017-01-25 14:06:15
言語 D
(dmd 2.109.1)
結果
AC  
実行時間 2 ms / 5,000 ms
コード長 1,085 bytes
コンパイル時間 657 ms
コンパイル使用メモリ 110,168 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-06-12 06:34:36
合計ジャッジ時間 1,507 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std.algorithm, std.conv, std.range, std.stdio, std.string;
import std.bitmanip;  // BitArray
import std.container;

void main()
{
  auto n = readln.chomp.to!size_t;
  auto di = readln.split.to!(int[]);
  auto wi = readln.split.to!(int[]).to!(bool[]);

  auto aij = new bool[][](n, n + 1);
  foreach (i; n.iota) {
    auto ii = i.to!int, ni = n.to!int;
    auto d1 = (ii + di[i]) % ni;
    auto d2 = ((ii - di[i]) % ni + ni) % ni;
    aij[d1][i] = aij[d2][i] = true;
    aij[i][n] = !wi[i];
  }

  gaussJordan(aij, n);
  writeln(hasSolution(aij, n) ? "Yes" : "No");
}

void gaussJordan(bool[][] aij, size_t n)
{
  size_t i, j;
  while (i < n && j < n) {
    if (!aij[i][j]) {
      auto k = aij[i+1..$].countUntil!(ai => ai[j]);
      if (k == -1)
        ++j;
      else
        swap(aij[i], aij[i + k + 1]);
    }

    if (aij[i][j]) {
      foreach (k, ref ai; aij[i+1..$])
        if (ai[j]) ai[] ^= aij[i][];
      ++i;
      ++j;
    }
  }
}

auto hasSolution(bool[][] aij, size_t n)
{
  foreach (ai; aij)
    if (ai[0..n].all!"!a" && ai[n]) return false;
  return true;
}
0