結果

問題 No.3426 Mod K Graph Increments (Hard)
コンテスト
ユーザー tomerun
提出日時 2026-01-11 15:46:44
言語 Crystal
(1.18.2)
結果
AC  
実行時間 334 ms / 2,000 ms
コード長 1,211 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 15,473 ms
コンパイル使用メモリ 335,704 KB
実行使用メモリ 40,612 KB
最終ジャッジ日時 2026-01-11 15:47:01
合計ジャッジ時間 15,442 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 10
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

read_line.to_i.times do
  puts solve() ? "Yes" : "No"
end

def solve
  n, m, k = read_line.split.map(&.to_i)
  g = Array.new(n) { [] of Int32 }
  m.times do
    u, v = read_line.split.map { |v| v.to_i - 1 }
    g[u] << v
    g[v] << u
  end
  b = read_line.split.map(&.to_i)
  g2 = Array.new(n) { [] of Int32 }
  visited = Array.new(n, false)
  visited[0] = true
  parent = Array.new(n, -1)
  q = [0]
  qi = 0
  while qi < n
    cur = q[qi]
    qi += 1
    g[cur].each do |adj|
      next if visited[adj]
      g2[cur] << adj
      g2[adj] << cur
      parent[adj] = cur
      visited[adj] = true
      q << adj
    end
  end
  qi = n - 1
  while qi > 0
    cur = q[qi]
    qi -= 1
    b[parent[cur]] += k - b[cur]
    b[parent[cur]] %= k
    b[cur] = 0
  end
  if b.all? { |v| v == 0 }
    return true
  end

  color = Array.new(n, 0)
  color[0] = 1
  q = [0]
  qi = 0
  ok = false
  while qi < n
    cur = q[qi]
    qi += 1
    g[cur].each do |adj|
      if color[adj] == color[cur]
        ok = true
      end
      next if color[adj] != 0
      color[adj] = -color[cur]
      q << adj
    end
  end
  if !ok
    return false
  end

  if k % 2 == 0
    return b.sum % 2 == 0
  else
    return true
  end
end
0