# クラスカル 法で「最小全域木」を求める N, M, K = gets.split.map(&:to_i) loads = Array.new(M) M.times{ |i| loads[i] = gets.split.map(&:to_i) } K.times{ i = gets.to_i - 1 loads[i][2] = 0 # 必ず選択しないといけないloadはコスト0とする。 } # コスト順にソート loads = loads.sort_by{ |load| load[2]} forest = [] NOT_FOUND = -1 ignored_cost = 0 for path in loads do from, to, c = path f_tree = NOT_FOUND t_tree = NOT_FOUND forest.size.times{ |i| # from, toの属するtreeを探す if forest[i].include?(from) then f_tree = i end if forest[i].include?(to) then t_tree = i end if f_tree != NOT_FOUND && t_tree != NOT_FOUND then break end } if f_tree != NOT_FOUND && t_tree == NOT_FOUND then # fromがあるtreeに属し、toがどのtreeにも属していない場合 # f_treeにtoを追加し、そのloadを採用する forest[f_tree].push(to) elsif t_tree != NOT_FOUND && f_tree == NOT_FOUND then # toがあるtreeに属し、fromがどのtreeにも属していない場合 # t_treeにfromを追加し、そのloadを採用する forest[t_tree].push(from) elsif t_tree != NOT_FOUND && f_tree != NOT_FOUND && t_tree != f_tree then # from, toどちらも異なるtreeに属している場合 # 2つのtreeを結合し、そのloadを採用する from_tree = forest[f_tree] to_tree = forest[t_tree] forest = forest.select{ |t| t != to_tree && t != from_tree}.push(from_tree + to_tree) elsif t_tree == NOT_FOUND && f_tree == NOT_FOUND then # from, toどちらもどのtreeにも属していない場合 # 新たなtreeとしてforestに追加し、そのloadを採用する forest.push([from, to]) else # from, toどちらも同じtreeに属する場合、そのloadは破棄する ignored_cost += c end end puts ignored_cost