結果

問題 No.2563 色ごとのグループ
ユーザー ngng628ngng628
提出日時 2023-12-02 17:30:08
言語 Crystal
(1.11.2)
結果
AC  
実行時間 399 ms / 2,000 ms
コード長 2,189 bytes
コンパイル時間 14,768 ms
コンパイル使用メモリ 296,576 KB
実行使用メモリ 74,112 KB
最終ジャッジ日時 2024-09-26 21:14:16
合計ジャッジ時間 20,615 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 3 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 2 ms
5,376 KB
testcase_13 AC 2 ms
5,376 KB
testcase_14 AC 4 ms
5,376 KB
testcase_15 AC 4 ms
5,376 KB
testcase_16 AC 3 ms
5,376 KB
testcase_17 AC 4 ms
5,376 KB
testcase_18 AC 3 ms
5,376 KB
testcase_19 AC 6 ms
5,376 KB
testcase_20 AC 19 ms
8,832 KB
testcase_21 AC 14 ms
5,888 KB
testcase_22 AC 9 ms
5,376 KB
testcase_23 AC 13 ms
5,760 KB
testcase_24 AC 132 ms
31,360 KB
testcase_25 AC 122 ms
33,152 KB
testcase_26 AC 184 ms
49,920 KB
testcase_27 AC 217 ms
62,464 KB
testcase_28 AC 219 ms
55,936 KB
testcase_29 AC 285 ms
66,688 KB
testcase_30 AC 291 ms
73,600 KB
testcase_31 AC 289 ms
65,792 KB
testcase_32 AC 288 ms
66,560 KB
testcase_33 AC 397 ms
74,112 KB
testcase_34 AC 368 ms
71,936 KB
testcase_35 AC 380 ms
71,936 KB
testcase_36 AC 399 ms
69,632 KB
testcase_37 AC 391 ms
66,944 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

main = -> {
  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|
    sz = sizes.fetch(color, 0_i64)
    next 0_i64 if sz == 0
    ut = NgLib::DisjointSet.new(sz)
    graph[color].each_with_index do |edges, u|
      edges.each do |v|
        ut.unite(u, v)
      end
    end
    (0...sz).count { |i| ut.leader(i) == i }.to_i64 - 1
  }
  
  puts ans
}

main.call

module NgLib
  class DisjointSet
    @n : Int32
    @parent_or_size : Array(Int32)
  
    def initialize
      @n = 0
      @parent_or_size = Array(Int32).new
    end
  
    def initialize(size : Int)
      @n = size.to_i32
      @parent_or_size = [-1] * size
    end
  
    def unite(a : Int, b : Int) : Int32
      x = leader(a)
      y = leader(b)
      return x if x == y
      if -@parent_or_size[x] < -@parent_or_size[y]
        x, y = y, x
      end
      @parent_or_size[x] += @parent_or_size[y]
      @parent_or_size[y] = x
      x
    end
  
    def equiv?(a : Int, b : Int) : Bool
      leader(a) == leader(b)
    end
  
    def leader(a : Int) : Int32
      return a.to_i32 if @parent_or_size[a] < 0
      @parent_or_size[a] = leader(@parent_or_size[a])
      @parent_or_size[a]
    end
  
    def size(a : Int) : Int32
      -@parent_or_size[leader(a)]
    end
  
    def groups : Array(Array(Int32)) | Nil
      leader_buf = [0] * @n
      group_size = [0] * @n
      @n.times do |i|
        leader_buf[i] = leader(i)
        group_size[leader_buf[i]] += 1
      end
      res = Array.new(@n){ [] of Int32 }
      @n.times do |i|
        res[leader_buf[i]] << i
      end
      res.delete([] of Int32)
      res
    end
  end
end
0