Submission #1334290

---

Source Code Expand

fn getline() -> String{
    let mut __ret = String::new();
    std::io::stdin().read_line(&mut __ret).ok();
    return __ret;
}

struct UnionFind {
    par: Vec<usize>,
    rank: Vec<usize>,
}

impl UnionFind {
    fn new(n: usize) -> UnionFind {
        let mut vec = vec![0;n];
        for i in 0..n {
            vec[i] = i;
        }
        UnionFind {
            par : vec,
            rank : vec![0;n],
        }
    }

    fn find(&mut self, x: usize) -> usize {
        if x == self.par[x] {
            x
        }else{
            let par = self.par[x];
            let res = self.find(par);
            self.par[x] = res;
            res
        }
    }

    fn same(&mut self, a: usize, b: usize) -> bool {
        self.find(a) == self.find(b)
    }

    fn unite(&mut self, a: usize, b: usize){
        let apar = self.find(a);
        let bpar = self.find(b);
        if self.rank[apar] > self.rank[bpar] {
            self.par[bpar] = apar;
        }else{
            self.par[apar] = bpar;
            if self.rank[apar] == self.rank[bpar] {
                self.rank[bpar] += 1;
            }
        }
    }
}

fn printvec<T: std::fmt::Display>(x: &Vec<T>){
    for v in x {
        print!("{} ",v);
    }
    println!("");
}

fn main() {
    let s = getline();
    let inp:Vec<_> = s.trim().split(' ').collect();
    let n: usize = inp[0].parse().unwrap();
    let q: usize = inp[1].parse().unwrap();
    let mut uf = UnionFind::new(n);
    // printvec(&uf.par);
    for _ in 0..q {
        let s = getline();
        let inp:Vec<_> = s.trim().split(' ').collect();
        let p: i32 = inp[0].parse().unwrap();
        let a: usize = inp[1].parse().unwrap();
        let b: usize = inp[2].parse().unwrap();
        if p == 0 {
            uf.unite(a,b);
            // printvec(&(uf.par));
        }else{
            if uf.same(a,b) {
                println!("Yes");
            }else{
                println!("No");
            }
        }
    }
}

Submission Info

Compile Error

warning: function is never used: `printvec`, #[warn(dead_code)] on by default
  --> ./Main.rs:54:1
   |
54 |   fn printvec<T: std::fmt::Display>(x: &Vec<T>){
   |  _^ starting here...
55 | |     for v in x {
56 | |         print!("{} ",v);
57 | |     }
58 | |     println!("");
59 | | }
   | |_^ ...ending here

Judge Result