rustgym 0.2.0

rustgym solutions
Documentation 
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
41
42
43
44

struct Solution;

impl Solution {
    fn combine(n: i32, k: i32) -> Vec<Vec<i32>> {
        let mut res = vec![];
        let nums: Vec<i32> = (1..=n).collect();
        let n = n as usize;
        let k = k as usize;
        let mut cur: Vec<i32> = vec![];
        Self::dfs(0, 0, &mut cur, &mut res, &nums, n, k);
        res
    }

    fn dfs(
        start: usize,
        count: usize,
        cur: &mut Vec<i32>,
        all: &mut Vec<Vec<i32>>,
        nums: &[i32],
        n: usize,
        k: usize,
    ) {
        if count == k {
            all.push(cur.to_vec());
        } else {
            for i in start..n {
                cur.push(nums[i]);
                Self::dfs(i + 1, count + 1, cur, all, nums, n, k);
                cur.pop();
            }
        }
    }
}

#[test]
fn test() {
    let n = 4;
    let k = 2;
    let mut res = vec_vec_i32![[2, 4], [3, 4], [2, 3], [1, 2], [1, 3], [1, 4]];
    let mut ans = Solution::combine(n, k);
    res.sort();
    ans.sort();
    assert_eq!(ans, res);
}