Leetcode 216. 组合总和 III

找出所有相加之和为 n 的 个数的组合组合中只允许含有 1 - 9 的正整数,并且每种组合中不存在重复的数字。

说明:

  • 所有数字都是正整数。
  • 解集不能包含重复的组合。 

示例 1:

输入: k = 3, n = 7
输出: [[1,2,4]]

示例 2:

输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]

 

基本思路:像这种要把所有可能输出的情况,想都不想直接递归解。

怎么才能递归呢?很简单,对于一个数,要它和不要它有两种选择,那么用着两种方式来进行递归即可。

 

class Solution {
public:
	vector> Res;
	vector> combinationSum3(int k, int n) {
		int N = n / k;
		vector Temp;
		_combinationSum3(k, n, Temp, 1);
		return Res;
	}

	/*** k 和 n就不解释了就是两个输入,Now_P表示的是当前找到哪个数了,它是从1开始逐步递增,2,3,4,……  */
	void _combinationSum3(int k, int n, vector Temp, int Now_P) {
		int Sum = 0;
		for (int n : Temp) Sum += n;

		/** if如果成立 就表示找到一个符合条件的数组了,将其加入到结果中 **/
		if (Temp.size() == k - 1 && Sum + Now_P == n) {
			Temp.push_back(Now_P);
			Res.push_back(Temp);
			return;
		}
		/** if没成立 就表示还没找到一个符合条件的数组,继续寻找 **/
		else {
			if (Sum + Now_P>n || Now_P>n|| Now_P>=9)
				return;
			_combinationSum3(k, n, Temp, Now_P + 1); // 不要当前的数
			Temp.push_back(Now_P);
			_combinationSum3(k, n, Temp, Now_P + 1); // 要当前的数
		}
	}
};

 

你可能感兴趣的:(Leetcode)