Submission #41742673

---

Source Code Expand

import Foundation

func main() {
  var scanner = Scanner()
  let N = scanner.read(Int.self)
  var degree = [Int](repeating: 0, count: N)
  for _ in 0 ..< N - 1 {
    let u = scanner.read(Int.self) - 1
    let v = scanner.read(Int.self) - 1
    degree[u] += 1
    degree[v] += 1
  }
  var answer: [Int] = []
  var n = N
  for i in 0 ..< N {
    if degree[i] >= 3 {
      answer.append(degree[i])
      n -= degree[i] + 1
    }
  }
  while n > 0 {
    answer.append(2)
    n -= 3
  }
  answer.sort()
  print(answer.map(\.description).joined(separator: " "))
}

main()

struct Scanner {
  private var tokens = [String]()
  private var index = 0
  
  mutating func read<T>(_ type: T.Type) -> T where T: LosslessStringConvertible {
    T(read())!
  }
  
  mutating func read<T>(_ type: T.Type, count n: Int) -> [T] where T: LosslessStringConvertible {
    (0 ..< n).map { _ in read(type) }
  }
  
  private mutating func read() -> String {
    while tokens.count == index {
      tokens = readLine()!.split(separator: " ").map(\.description)
      index = 0
    }
    defer {
      index += 1
    }
    return tokens[index]
  }
}

Submission Info

Judge Result