【面试题】给一个数组和目标值，求数组中和为目标值的所有组合

最新推荐文章于 2023-06-27 09:00:00 发布

原创最新推荐文章于 2023-06-27 09:00:00 发布 · 1.2w 阅读

40 ·

CC 4.0 BY-SA版权

文章标签：

#数组求和组合 #目标值

面试准备同时被 2 个专栏收录

66 篇文章

订阅专栏

J2EE

21 篇文章

订阅专栏

博客围绕数组求和组合问题展开，给定数组和目标值，求数组中和为目标值的所有组合。分两种情况，一是数组中每个值可重复使用，二是值不能重复使用。解题思路采用深度优先遍历，值不可重复时从根节点之后遍历。

1.给一个数组和目标值，求数组中和为目标值的所有组合，数组中每个值可以重复使用。

比如：数组【2，3，6，7】，目标值7

所有组合：【2，2，3】，【7】

思路：用深度优先遍历来做。

public class Solution {
    ArrayList<ArrayList<Integer>> list = new ArrayList<>();
    public ArrayList<ArrayList<Integer>> combinationSum(int[] num, int target) {
		ArrayList<Integer> ls = new ArrayList<>();
		if (num == null || num.length == 0)
			return list;
        Arrays.sort(num);
		dfs(num, 0, target, ls);
        return list;
	}

	private void dfs(int[] arr, int index, int sum, ArrayList<Integer> ls) {
		if (sum < 0)
			return;
		if (sum == 0) {
			if (!list.contains(ls))
				list.add(new ArrayList<>(ls));
			return;
		}

		for (int i = index; i < arr.length; i++) {
			ls.add(arr[i]);
			dfs(arr, i, sum - arr[i], ls);
			ls.remove(ls.size() - 1);
		}

	}
}

2.当值不能重复使用时，只需要每次从根节点之后遍历即可。

public class Solution {
   ArrayList<ArrayList<Integer>> list = new ArrayList<>();

	public ArrayList<ArrayList<Integer>> combinationSum2(int[] num, int target) {
		ArrayList<Integer> ls = new ArrayList<>();
		if (num == null || num.length == 0)
			return list;
		Arrays.sort(num);
		dfs(num, 0, target, ls);
		return list;

	}

	private void dfs(int[] arr, int index, int sum, ArrayList<Integer> ls) {
		if (sum < 0)
			return;
		if (sum == 0) {
			if (!list.contains(ls))
				list.add(new ArrayList<>(ls));
			return;
		}

		for (int i = index; i < arr.length; i++) {
			ls.add(arr[i]);
            //此处从i+1开始
			dfs(arr, i + 1, sum - arr[i], ls);
			ls.remove(ls.size() - 1);
		}

	}
}