LeetCode 39. Combination Sum II

stephen_wong

于 2014-06-27 02:02:09 发布

阅读量758

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode LeetCode Problems 文章标签： LeetCode C++

本文链接：https://blog.youkuaiyun.com/stephen_wong/article/details/35043479

LeetCode 同时被 2 个专栏收录

186 篇文章

订阅专栏

LeetCode Problems

183 篇文章

订阅专栏

本文介绍了一种解决组合总和问题的算法实现。通过对输入的候选数字进行排序和去重处理，并使用递归方法搜索所有可能的组合来达到目标值。此算法适用于解决具有重复数字的组合问题。

和LeetCode 38. Combination Sum类似。

将candidates排升序后去重并用一个cnt数组记录对应数字的个数，再递归求解。

代码：

class Solution 
{
public:
	vector<vector<int> > combinationSum2(vector<int> &candidates, int target) 
	{
		sort(candidates.begin(), candidates.end());
		for (size_t i = 0; i < candidates.size(); ++ i)
		{
			if (i==0 || candidates[i]!=candidates[i-1])
			{
				num.push_back( candidates[i] );
				cnt.push_back( 1 );
			} else
			{
				++ cnt[cnt.size()-1];
			}
		}
		vector<int> now;
		dfs(now, 0, target);
		return ret;
	}
private:
	vector<int> num;
	vector<int> cnt;
	vector<vector<int> > ret;
	void dfs(vector<int>& now, size_t index, int remain)
	{
		for (size_t i = index; i < num.size(); ++ i)
		{
			if (num[i] > remain)
			{
				break;
			} else
			{
				int cur = remain;
				for (int j = 0; j<cnt[i] && num[i]<=cur; ++ j)
				{
					cur -= num[i];
					now.push_back( num[i] );
					dfs(now, i+1, cur);
				}
				if (cur == 0)
				{
					ret.push_back( now );
				}
				while (cur != remain)
				{
					now.pop_back();
					cur += num[i];
				}
			}
		}
	}
};