結果

問題 No.748 yuki国のお財布事情
ユーザー manhattanermanhattaner
提出日時 2018-10-31 22:55:06
言語 Ruby
(3.3.0)
結果
TLE  
実行時間 -
コード長 2,020 bytes
コンパイル時間 83 ms
コンパイル使用メモリ 7,680 KB
実行使用メモリ 38,688 KB
最終ジャッジ日時 2024-04-30 02:21:20
合計ジャッジ時間 8,597 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 90 ms
17,536 KB
testcase_01 AC 86 ms
12,160 KB
testcase_02 AC 89 ms
12,288 KB
testcase_03 AC 86 ms
12,160 KB
testcase_04 AC 85 ms
12,032 KB
testcase_05 AC 86 ms
12,288 KB
testcase_06 AC 85 ms
12,032 KB
testcase_07 AC 87 ms
12,032 KB
testcase_08 AC 87 ms
12,032 KB
testcase_09 AC 84 ms
12,288 KB
testcase_10 AC 85 ms
12,288 KB
testcase_11 AC 85 ms
12,288 KB
testcase_12 AC 86 ms
12,160 KB
testcase_13 AC 245 ms
13,824 KB
testcase_14 AC 1,856 ms
38,400 KB
testcase_15 AC 378 ms
14,592 KB
testcase_16 TLE -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff #

# クラスカル 法で「最小全域木」を求める

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
0