結果

問題 No.2563 色ごとのグループ
コンテスト
ユーザー ngng628
提出日時 2023-12-02 15:28:44
言語 Crystal
(1.14.0)
結果
WA  
実行時間 -
コード長 727 bytes
コンパイル時間 14,221 ms
コンパイル使用メモリ 295,300 KB
実行使用メモリ 73,472 KB
最終ジャッジ日時 2024-09-26 18:45:11
合計ジャッジ時間 17,293 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30 WA * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

n, m = read_line.split.map &.to_i64
c = read_line.split.map &.to_i64.pred
sizes = c.tally
graph = Array.new(n) { |color| Array(Array(Int64)).new(sizes[color]? || 0) { [] of Int64 } }
hash = Array.new(n) { Hash(Int64, Int32).new }
m.times do
  u, v = read_line.split.map &.to_i64.pred
  if c[u] == c[v]
    color = c[u]
    hash[color][u] = hash[color].size unless hash[color].has_key?(u)
    hash[color][v] = hash[color].size unless hash[color].has_key?(v)
    u2 = hash[color][u]
    v2 = hash[color][v]
    graph[color][u2] << v2
    graph[color][v2] << u2
  end
end

ans = n.times.sum { |color|
  n_edges = graph[color].sum(&.size.to_i64) // 2
  Math.max(0_i64, (sizes.fetch(color, 0_i64).to_i64 - 1) - n_edges)
}

puts ans
0