我理解的算法 - 四数之和
LeetCode 18.四数之和
这道题的题目大家自行查看:链接在这
其解法和思路与两数之和和三数之和都是一样的,还不熟悉的小朋友们可以看这里:两数之和,三数之和
那么这道题目的难点在哪里呢?其实这道题目的难点就在于要求的和变成了四个数,比三个数还多一个数,所以我们要想办法尽量把多余的一些处理给它优化掉,这样才能提高其效率,我们来看一下吧
四数之和的思路和三数之和是一样的,我们还是可以使用双指针来做,只不过这边我们需要固定2个元素即固定a和b来移动c和d,来看他们的和是比目标值大还是小,这样就会比三数之和多一次循环。代码如下
public List<List<Integer>> fourSum(int[] nums, int target) {
List<List<Integer>> retResult = new ArrayList<>();
int length = nums.length;
if (nums == null || length < 4) {
return retResult;
}
Arrays.sort(nums);//排序,为了下面去重
for(int i = 0; i < length - 3; i++){
if(i > 0 && nums[i] == nums[i - 1]) continue;//去重
if((long) nums[i] + nums[i + 1] + nums[i + 2] + nums[i + 3] > target) break;//剪枝
if((long) nums[i] + nums[length - 3] + nums[length - 2] + nums[length - 1] < target) continue;//剪枝
for(int j = i + 1; j < length - 2; j++){
if(j > i + 1 && nums[j] == nums[j - 1]) continue;//去重
if((long) nums[i] + nums[j] + nums[j + 1] + nums[j + 2] > target) break;//剪枝
if ((long) nums[i] + nums[j] + nums[length - 2] + nums[length - 1] < target) continue;//剪枝
int left = j + 1;
int right = length - 1;
while(left < right){
int diff = nums[i] + nums[j] + nums[left] + nums[right] - target;
if(diff > 0){
while(left < right && nums[right] == nums[--right]);//去重
}else if(diff < 0){
while(left < right && nums[left] == nums[++left]);//去重
}else{
retResult.add(new ArrayList<>(Arrays.asList(nums[i], nums[j], nums[left], nums[right])));
while(left < right && nums[right] == nums[--right]);//去重
while(left < right && nums[left] == nums[++left]);//去重
}
}
}
}
return retResult;
}
这边需要说明一下那些剪枝的操作,剪枝就是说,如果我们可以判断出来后面的数没必要再进行计算了,那我们就可以跳过这些步骤,从而降低执行时间,就像把没有用的树枝剪掉一样。
第一段剪枝操作
if((long) nums[i] + nums[i + 1] + nums[i + 2] + nums[i + 3] > target) break;//剪枝
这边就是 a + b + c + d a+b+c+d a+b+c+d如果大于0了,那说么在d+1往后肯定都是大于0的了,所以我们就没必要再进行后续的计算了,直接break掉。
第二段
if((long) nums[i] + nums[length - 3] + nums[length - 2] + nums[length - 1] < target) continue;//剪枝
如果我们假设用最大的三个数加上了此次固定的a还比目标数小,那么就说明我们的a肯定小了,所以需要继续遍历而不需要再执行移动双指针来尝试了。
下面2段和上面的逻辑是一样的,这样做后,可以大幅度的缩减执行时间,提高执行效率,精华在剪枝处理上,这样的算法你理解了吗