Submission #423429

---

Source Code Expand

import scala.collection.mutable._


object Main {

  class UnionFind(val n: Int) {
    var par = MutableList.range(0,n)
    var rank = MutableList.fill(n)(0)

    def root(x: Int): Int =
      if(par(x)==x) x
      else {par(x) = root(par(x)); par(x)}


    def same(x: Int, y: Int): Boolean = root(x)==root(y)

    def unite(ax: Int, ay: Int): Unit = {
      val x = root(ax)
      val y = root(ay)
      if(x!=y){
        if(rank(x) < rank(y)) par(x) = y
        else {
          par(y) = x
          if(rank(x) == rank(y))rank(x) = rank(x)+1;
        }
      }
    }
  }

  def main(args: Array[String]): Unit = {
    val Array(n,q,_*) = readLine.split(" ").map(_.toInt)
    val uf = new UnionFind(n)
    for(i <- 1 to q){
      val Array(p,a,b,_*) = readLine.split(" ").map(_.toInt)
      p match {
        case 0 => uf.unite(a,b)
        case 1 if uf.same(a,b) => println("Yes")
        case 1 => println("No")
      }
    }
  }
}

Submission Info

Compile Error

warning: there were two deprecation warnings; re-run with -deprecation for details
one warning found

Judge Result