solve 39

Author: Zhang Xiaodong

src/solution/mod.rs

@@ -34,3 +34,4 @@ mod s0035_search_insert_position;
 mod s0036_valid_sudoku;
 mod s0037_sudoku_solver;
 mod s0038_count_and_say;
+mod s0039_combination_sum;

src/solution/s0039_combination_sum.rs

@@ -0,0 +1,105 @@
+/**
+ * [39] 组合总和
+ *
+ * 给你一个 无重复元素 的整数数组 candidates 和一个目标整数 target ，找出 candidates 中可以使数字和为目标数 target 的 所有 不同组合 ，并以列表形式返回。你可以按 任意顺序 返回这些组合。
+ * candidates 中的 同一个 数字可以 无限制重复被选取 。如果至少一个数字的被选数量不同，则两种组合是不同的。
+ * 对于给定的输入，保证和为 target 的不同组合数少于 150 个。
+ *  
+ * 示例 1：
+ *
+ * 输入：candidates = [2,3,6,7], target = 7
+ * 输出：[[2,2,3],[7]]
+ * 解释：
+ * 2 和 3 可以形成一组候选，2 + 2 + 3 = 7 。注意 2 可以使用多次。
+ * 7 也是一个候选， 7 = 7 。
+ * 仅有这两种组合。
+ * 示例 2：
+ *
+ * 输入: candidates = [2,3,5], target = 8
+ * 输出: [[2,2,2,2],[2,3,3],[3,5]]
+ * 示例 3：
+ *
+ * 输入: candidates = [2], target = 1
+ * 输出: []
+ *
+ *  
+ * 提示：
+ *
+ * 	1 <= candidates.length <= 30
+ * 	2 <= candidates[i] <= 40
+ * 	candidates 的所有元素 互不相同
+ * 	1 <= target <= 40
+ *
+ */
+pub struct Solution {}
+
+// problem: https://leetcode.cn/problems/combination-sum/
+// discuss: https://leetcode.cn/problems/combination-sum/discuss/?currentPage=1&orderBy=most_votes&query=
+
+// submission codes start here
+
+impl Solution {
+    pub fn combination_sum(candidates: Vec<i32>, target: i32) -> Vec<Vec<i32>> {
+        let mut candidates = candidates.clone();
+        candidates.sort();
+
+        let mut res = vec![];
+
+        let mut curr = vec![];
+        let mut curr_idx = vec![];
+        let mut sum = 0;
+
+        curr.push(candidates[0]);
+        curr_idx.push(0);
+        sum += candidates[0];
+
+        while curr.len() > 0 {
+            // println!("{:?} {} {}", curr, sum, target);
+            if sum < target {
+                let aidx = curr_idx[curr_idx.len()-1];
+                curr.push(candidates[aidx]);
+                curr_idx.push(aidx);
+                sum += candidates[aidx];
+                continue;    
+            }
+            if sum == target {
+                res.push(curr.clone());
+            }
+            while let Some(i) = curr_idx.pop() {
+                let v = curr.pop().unwrap();
+                sum -= v;
+
+                if i + 1 < candidates.len() {
+                    let i: usize = i + 1;
+                    curr.push(candidates[i]);
+                    curr_idx.push(i);
+                    sum += candidates[i];
+                    break;
+                }
+            }
+        }
+        res
+    }
+}
+
+// submission codes end
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_39() {
+        let empty_res: Vec<Vec<i32>> = vec![];
+
+        assert_eq!(
+            Solution::combination_sum(vec![2, 3, 6, 7], 7),
+            vec![vec![2, 2, 3], vec![7]]
+        );
+        assert_eq!(Solution::combination_sum(vec![2], 1), empty_res);
+        assert_eq!(
+            Solution::combination_sum(vec![2, 3, 5], 8),
+            vec![vec![2, 2, 2, 2], vec![2, 3, 3], vec![3, 5]]
+        );
+    }
+}