結果

問題 No.1640 簡単な色塗り
コンテスト
ユーザー 箱星
提出日時 2021-08-06 22:51:41
言語 Kotlin
(2.1.0)
結果
TLE  
実行時間 -
コード長 2,111 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 12,720 ms
コンパイル使用メモリ 459,148 KB
実行使用メモリ 146,408 KB
最終ジャッジ日時 2024-06-29 15:56:39
合計ジャッジ時間 19,702 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 2 TLE * 1 -- * 50
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import java.io.PrintWriter
import java.util.*
import kotlin.math.*

fun PrintWriter.solve() {
    val n = nextInt()
    val m = n
    val adj = Array(n) { mutableListOf<Int>() }
    val edges = mutableMapOf<Pair<Int, Int>, Stack<Int>>()
    for (i in 0 until m) {
        val v1 = nextInt() - 1
        val v2 = nextInt() - 1
        adj[v1].add(v2)
        adj[v2].add(v1)
        val pair = min(v1, v2) to max(v1, v2)
        if (!edges.containsKey(pair)) {
            edges[pair] = Stack()
        }
        edges[pair]?.push(i)
    }
    val choice = Array(n) { -1 }
    val visited = BooleanArray(n) { false }
    val chosen = BooleanArray(n) { false }
    for (i in 0 until n) {
        if (visited[i]) continue
        val stack = Stack<Pair<Int, Int>>()
        stack.push(i to -1)
        while (stack.isNotEmpty()) {
            val (v, p) = stack.pop()
            for (w in adj[v]) {
                if (w != p && w != v) {
                    stack.push(w to v)
                }
                val min = min(v, w)
                val max = max(v, w)
                if (edges[min to max]?.isNotEmpty()!!) {
                    val num = edges[min to max]?.pop()!!
                    if (!chosen[min]) {
                        chosen[min] = true
                        choice[num] = min
                    } else {
                        chosen[max] = true
                        choice[num] = max
                    }
                }
            }
        }
    }
    if (chosen.any { !it }) {
        println("No")
    } else {
        println("Yes")
        println(choice.map { it + 1 }.joinToString("\n"))
    }
}

fun main() {
    val writer = PrintWriter(System.out, false)
    writer.solve()
    writer.flush()
}

// region Scanner
private var st = StringTokenizer("")
private val br = System.`in`.bufferedReader()

fun next(): String {
    while (!st.hasMoreTokens()) st = StringTokenizer(br.readLine())
    return st.nextToken()
}

fun nextInt() = next().toInt()
fun nextLong() = next().toLong()
fun nextLine() = br.readLine()
fun nextDouble() = next().toDouble()
// endregion
0