結果

問題 No.748 yuki国のお財布事情
ユーザー manhattanermanhattaner
提出日時 2018-10-31 22:55:06
言語 Ruby
(3.3.0)
結果
TLE  
実行時間 -
コード長 2,020 bytes
コンパイル時間 857 ms
コンパイル使用メモリ 7,680 KB
実行使用メモリ 45,476 KB
最終ジャッジ日時 2024-11-19 13:43:27
合計ジャッジ時間 34,512 ms
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 77 ms
19,104 KB
testcase_01 AC 75 ms
43,776 KB
testcase_02 AC 74 ms
18,980 KB
testcase_03 AC 77 ms
40,192 KB
testcase_04 AC 77 ms
19,104 KB
testcase_05 AC 80 ms
33,280 KB
testcase_06 AC 80 ms
19,232 KB
testcase_07 AC 78 ms
32,896 KB
testcase_08 AC 78 ms
19,236 KB
testcase_09 AC 79 ms
32,640 KB
testcase_10 AC 76 ms
19,108 KB
testcase_11 AC 76 ms
19,108 KB
testcase_12 AC 77 ms
18,980 KB
testcase_13 AC 224 ms
34,432 KB
testcase_14 AC 1,707 ms
45,476 KB
testcase_15 AC 341 ms
35,200 KB
testcase_16 TLE -
testcase_17 TLE -
testcase_18 TLE -
testcase_19 TLE -
testcase_20 TLE -
testcase_21 TLE -
testcase_22 AC 79 ms
12,160 KB
testcase_23 AC 81 ms
12,288 KB
testcase_24 AC 79 ms
12,288 KB
testcase_25 AC 650 ms
20,480 KB
testcase_26 TLE -
testcase_27 TLE -
testcase_28 TLE -
権限があれば一括ダウンロードができます
コンパイルメッセージ
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