結果

問題 No.1059 素敵な集合
ユーザー yudedako
提出日時 2020-08-27 10:48:16
言語 Kotlin
(2.1.0)
結果
AC  
実行時間 393 ms / 2,000 ms
コード長 1,239 bytes
コンパイル時間 13,619 ms
コンパイル使用メモリ 451,472 KB
実行使用メモリ 57,604 KB
最終ジャッジ日時 2024-07-23 09:41:38
合計ジャッジ時間 22,744 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 


class UnionFind(val size: Int) {
    private val vec = IntArray(size){-1}
    fun find(a: Int): Int {
        return if (vec[a] < 0) {
            a
        }else {
            vec[a] = find(vec[a])
            vec[a]
        }
    }
    fun same(a: Int, b: Int): Boolean {
        return find(a) == find(b)
    }
    fun unite(a: Int, b: Int): Boolean {
        val ar = find(a)
        val br = find(b)
        if (ar == br) return false
        if (vec[ar] < vec[br]) {
            vec[ar] += vec[br]
            vec[br] = ar
        }else {
            vec[br] += vec[ar]
            vec[ar] = br
        }
        return true
    }
    fun sizeOf(a: Int): Int {
        return -vec[find(a)]
    }
}

fun main() {
    val (left, right) = readLine()!!.trim().split(' ').map(String::toInt)
    val uft = UnionFind(right - left + 1)
    var cost = 0
    outer@for (rest in 0 .. right) {
        for (i in left .. right) {
            for (j in i + rest .. right step i) {
                if (uft.unite(i - left, j - left)) {
                    cost += rest
                    if (uft.sizeOf(0) == uft.size) {
                        break@outer
                    }
                }
            }
        }
    }
    println(cost)
}
0