LeetCode216. Combination Sum III

最新推荐文章于 2023-12-07 14:18:24 发布

原创最新推荐文章于 2023-12-07 14:18:24 发布 · 328 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#c++ #leetcode #回溯

leetcode 专栏收录该内容

51 篇文章

订阅专栏

本文介绍LeetCode 216题“Combination Sum III”的解题思路及C++实现代码。该题旨在寻找所有可能的组合，这些组合由k个数构成并加总等于目标数n，且仅使用1到9的数字，每组组合中的数字不重复。

LeetCode216. Combination Sum III

题目：

Find all possible combinations of k numbers that add up to a number n, given that only numbers from 1 to 9 can be used and each combination should be a unique set of numbers.

Example 1:

Input: k = 3, n = 7

Output:

[[1,2,4]]

Example 2:

Input: k = 3, n = 9

Output:

[[1,2,6], [1,3,5], [2,3,4]]

题意分析：

这道题和LeetCode39基本用的是同一个思路。也是背包问题的扩展版。属于回溯的经典问题。

void isok(vector<vector<int>>& result, int n, int begin, vector<int>& tmp, int k) { //n:target
	if (n == 0 && tmp.size() == k) {
		result.push_back(tmp);
		return;
	}
	else {
		for (int i = begin; i <= 9; i++) {
			if (i > n) break;
			tmp.push_back(i);
			isok(result, n - i, i+1, tmp, k);
			tmp.pop_back();
		}
	}
}

代码：

class Solution {
public:
	vector<vector<int>> combinationSum3(int k, int n) {
		vector<vector<int>> result;
		vector<int> tmp;
		isok(result, n, 1, tmp, k);
		return result;
	}

	void isok(vector<vector<int>>& result, int n, int begin, vector<int>& tmp, int k) { //n:target
		if (n == 0 && tmp.size() == k) {
			result.push_back(tmp);
			return;
		}
		else {
			for (int i = begin; i <= 9; i++) {
				if (i > n) break;
				tmp.push_back(i);
				isok(result, n - i, i+1, tmp, k);
				tmp.pop_back();
			}
		}
	}
};