-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathlib.rs
40 lines (38 loc) · 930 Bytes
/
lib.rs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
///Find all unique subsets of a set with unique elements
///Iterative algorithm
pub fn subsets(arr: &[i32]) -> Vec<Vec<i32>> {
if arr.len() == 0 {
return vec![];
}
let mut subsets = Vec::<Vec<i32>>::with_capacity(1 << arr.len());
subsets.push(vec![]);
for elem in arr {
for i in 0..subsets.len() {
let subset = &subsets[i];
let mut new_subset = subset.to_vec();
new_subset.push(*elem);
subsets.push(new_subset);
}
}
subsets
}
#[cfg(test)]
mod tests {
use super::subsets;
#[test]
fn test_subsets() {
assert_eq!(
subsets(&[1, 2, 3]),
vec![
vec![],
vec![1],
vec![2],
vec![1, 2],
vec![3],
vec![1, 3],
vec![2, 3],
vec![1, 2, 3]
]
);
}
}