Problem:
D: Walking

問題

$1$ から $N$ の番号がつけられている $N$ 個の島がある.

それぞれの島は $N-1$ 個の橋によって、どの $2$ つの島も何本かの橋を渡って互いに移動することができる.

それぞれの橋には耐久度があり、入力が与えられた時点での $i$ 番目の橋の耐久度は $w_i$ である.

それぞれの島にはお宝が $1$ つずつ置いており、島に滞在しているときにお宝を拾うことができる.

現在島 $S$ にいるyebiくんは、島 $E$ にある博物館に全てのお宝を運びたい.

yebiくんは✝魔力✝を持っているので、島 $v$ に訪問するたびに、 $v$ から出る全ての橋の耐久度が $T$ 減少する.

橋の耐久度が $0$ 以下になったとき、橋は崩壊し、それ以降には渡ることができなくなる.

yebiくんは博物館に全てのお宝を届けることができるか？

ただし、yebiくんは力持ちなので同時にいくつでもお宝を持ち運ぶことができる.

制約

入力値は全て整数である

$2 \leq N \leq 10^5$

$1 \leq S, E \leq N$

$0 \leq T \leq 10^9$

$1 \leq w_i \leq 10^9$

$1 \leq a_i, b_i \leq N$

入力形式

入力は以下の形式で与えられる

$N\ T\ S\ E$

$a_1\ b_1\ w_1$

:

:

$a_{N-1}\ b_{N-1}\ w_{N-1}$

出力

博物館に全てのお宝を届けることができるなら "Yes"、そうでなければ "No" を出力せよ.

また、末尾に改行を出力せよ.

サンプル

サンプル入力 1

4 10 1 4
1 2 52
1 3 68
3 4 45

サンプル出力 1

Yes

サンプル入力 2

4 10 1 4
1 2 15
1 3 60
3 4 10

サンプル出力 2

No

サンプル入力 3

3 0 1 3
1 2 5
2 3 5

サンプル出力 3

Yes

yebiくんの魔力は貧弱すぎて、橋の耐久度を減らすことはできない.

Solution in Scala (judge verdict: Accepted)

import scala.collection.mutable
object Main extends App {
  import scala.io.StdIn._
  val Array(n, t, s, e) = readLine.trim.split(' ').map(_.toInt)
  val islands = Array.fill(n){Nil: List[(Int, Int)]}
  for (_ ← 1 until n){
    val Array(a, b, w) = readLine.trim.split(' ').map(_.toInt)
    islands(a - 1) ::= (b - 1, w)
    islands(b - 1) ::= (a - 1, w)
  }
  val bridgeFromS = Array.fill(n){-1}
  val queue = mutable.Queue[Int](s - 1)
  while(queue.nonEmpty) {
    val current = queue.dequeue()
    for ((next, _) ← islands(current) if next != s - 1 && bridgeFromS(next) == -1) {
      queue.enqueue(next)
      bridgeFromS(next) = current
    }
  }
  val bridgeWToE = Array.fill(n){0}
  val bridgeToE = Array.fill(n){-1}
  queue.enqueue(e - 1)
  while(queue.nonEmpty) {
    val current = queue.dequeue()
    for ((next, w) ← islands(current) if next != e - 1 && bridgeToE(next) == -1) {
      bridgeToE(next) = current
      bridgeWToE(next) = w
      queue.enqueue(next)
    }
  }
  val maxDamage = Array.fill(n){0}
  var canCollect = true
  for (current ← 0 until n if canCollect) {
    if (current == s - 1){
      canCollect = islands(current).filter(_._1 != bridgeToE(current)).map{case (i, w) ⇒ w - islands(i).length * t.toLong}.sorted.zipWithIndex.map{case (m, r) ⇒ m - r * t.toLong}.forall(_ > 0) && ((s == e) || bridgeWToE(current) > (islands(current).length - 1) * t.toLong)
    }else if (current == e - 1) {
      canCollect = islands(current).filter(_._1 != bridgeFromS(current)).map{case (i, w) ⇒ w - (islands(i).length + 1) * t.toLong}.sorted.zipWithIndex.map{case (m, r) ⇒ m - r * t.toLong}.forall(_ > 0)
    }else if (bridgeFromS(current) == bridgeToE(current)) {
      canCollect = islands(current).filter(_._1 != bridgeFromS(current)).map{case (i, w) ⇒ w - (islands(i).length + 1) * t.toLong}.sorted.zipWithIndex.map{case (m, r) ⇒ m - r * t.toLong}.forall(_ > 0)
    }else {
      canCollect = islands(current).filter(b ⇒ b._1 != bridgeFromS(current) && b._1 != bridgeToE(current)).map{case (i, w) ⇒ w - islands(i).length * t.toLong}.sorted.zipWithIndex.map{case (m, r) ⇒ m - (r + 1) * t.toLong}.forall(_ > 0) && bridgeWToE(current) > (islands(current).length - 1) * t.toLong
    }
  }
  println(
    if (canCollect) "Yes" else "No"
  )
}