还是前不久写的《四树之和》那题:Sort函数居然能改变元素值?记一次有趣的bug——四树之和
唉···dfs一直超时没过的去。于是我做了点剪枝优化后,卷土重来:
class Solution {
public:
vector<vector<int>>res;
vector<int> temp;
int sum;
void dfs(vector<int> nums, int sub)
{
if (temp.size() == 4)
{
if (sum == 0)
{
res.push_back(temp);
}
return;
}
if (sub >= nums.size())
{
return;
}
if((4-temp.size())*nums[sub]>-sum)
{
return;
}
temp.push_back(nums[sub]);
sum += nums[sub];
dfs(nums, sub + 1);
sum -= nums[sub];
temp.pop_back();
int next = sub+1;
while (next<nums.size()&&nums[next] == nums[sub]) { next++; }
dfs(nums, next);
}
vector<vector<int>> fourSum(vector<int>& nums, int target) {
sort(nums.begin(),nums.end());
sum = -target;
dfs(nums,0);
return res;
}
};
然鹅,每次运行 都只输出:
于是我反反复复对代码的逻辑进行复查,翘了三节课就在这儿找bug,结果找死了也找不到。
于是我掏出我的visual studio2019企业版。。。对每一次递归的值进行追踪。
发现
if((4-temp.size())*nums[sub]>-sum) { return; }
每一次这个if的条件都成立。
此刻我:??????????
进行了一番试验后,我感到我的脸上发出了啪啪啪的响声。。。。。。
原来————
temp.size()的数据类型是unsigned int ,当它与负数相乘时,就会出现奇怪的结果——一个巨大的整数。于是每次if的条件基本上都成立。
所以只要改成这样即可:
if((4-(int)temp.size())*nums[sub]>-sum) { return; }
这鸽了半个月的题目终于可以用回溯算法切掉了!(开心)
开nmmmmmm心