File tree Expand file tree Collapse file tree 1 file changed +6
-5
lines changed Expand file tree Collapse file tree 1 file changed +6
-5
lines changed Original file line number Diff line number Diff line change @@ -9,11 +9,12 @@ pub struct Dsu {
9
9
parent_or_size : Vec < i32 > ,
10
10
}
11
11
12
+
12
13
impl Dsu {
13
- pub fn new ( n : usize ) -> Self {
14
+ pub fn new ( size : usize ) -> Self {
14
15
Self {
15
- n : n ,
16
- parent_or_size : vec ! [ -1 ; n ] ,
16
+ n : size ,
17
+ parent_or_size : vec ! [ -1 ; size ] ,
17
18
}
18
19
}
19
20
pub fn merge ( & mut self , a : usize , b : usize ) -> usize {
@@ -28,13 +29,13 @@ impl Dsu {
28
29
}
29
30
self . parent_or_size [ x] += self . parent_or_size [ y] ;
30
31
self . parent_or_size [ y] = x as i32 ;
31
- return x ;
32
+ x
32
33
}
33
34
34
35
pub fn same ( & mut self , a : usize , b : usize ) -> bool {
35
36
assert ! ( a < self . n) ;
36
37
assert ! ( b < self . n) ;
37
- return self . leader ( a) == self . leader ( b) ;
38
+ self . leader ( a) == self . leader ( b)
38
39
}
39
40
pub fn leader ( & mut self , a : usize ) -> usize {
40
41
assert ! ( a < self . n) ;
You can’t perform that action at this time.
0 commit comments