在leetcode中 twoSum解决思路主要是排序的数组后,以升序为例,先从右端固定一个值Fx,然后从固定值的左邻居Ni开始遍历,如果固定值Fx+Ni=target,则break结束,否则,Fx向左移动一个值,同样思路左邻居顺序移动遍历,如果要求出所有的这种组合,那么Fx+Ni=target后,不执行break,直到Fx为倒数第二数为止。当然这里也可以优化,基本上固定一个值后左邻居Ni的值已经固定了大小,可以用二分法在数组中查询是否存在Ni这个值。
同样treeSum可以在twoSum基础上修改,即先固定一个值Fx将排序后固定值得左边子数组作为Twosum中数组,此时twoSum要求的目标值为target=target-Fx。
fourSum的实现原理可以基于treeSum。
总结下来其实就是一个深度搜索的问题,twoSum的搜索深度为2,treeSum的搜索深度为3,fourSum的搜索深度为4。如果求NSum问题,即搜索深度为4,所以总结下来实现思路如下:
class SolutionNSum {
public:
vector<vector<int> > permuteUnique(vector<int> &num,int target,int searchdpth) {
vector<vector<int> > res;
vector<int> out;
vector<int> visited(num.size(), 0);
sort(num.begin(), num.end());
permuteUniqueDFS(num, 0, target, searchdpth,visited, out, res);
return res;
}
void permuteUniqueDFS(vector<int> &num, int level, int target,int searchdpth,vector<int> &visited, vector<int> &out, vector<vector<int> > &res) {
if ((level >= num.size()|| target==0)&& searchdpth==0) res.push_back(out);
else if (target < 0) return;
else {
for (int i = level; i < num.size(); ++i) {
if (visited[i] == 0) {
if (i > 0 && num[i] == num[i - 1] && visited[i - 1] == 0) continue;
visited[i] = 1;
out.push_back(num[i]);
target = target- num[i];
if (target >= 0) {
permuteUniqueDFS(num, level + 1, target, searchdpth-1, visited, out, res);
}
target = target + num[i];
out.pop_back();
visited[i] = 0;
}
}
}
}
};
int main()
{
int target;
cin >> target;
string str;
vector<int> nums;
int data=0,i=0;
cin >> str;
while (i < str.size())
{
if (str[i] != ',')
{
data = data * 10 + str[i] - '0';
}
else
{
nums.push_back(data);
data = 0;
}
i++;
}
nums.push_back(data);
int n=nums.size();
SolutionNSum s3;
for (int searchdpth= 1; searchdpth < nums.size(); searchdpth++) {
vector<vector<int> > res = s3.permuteUnique(nums, target, searchdpth);
for (int i = 0; i < res.size(); i++)
{
if (res[i].size() == 1)
{
cout << target << "=" << res[i][0] << endl;
}
else
{
cout << target << "=" << res[i][0];
for (int j = 1; j < res[i].size(); j++)
cout << "+" << res[i][j];
cout << endl;
}
}
}
return 0;
}
测试结果如下:
第一行表示目标值,第二行为数组,接下来为查找结果。