力扣刷题 18.四数之和

目录

题干

解题思路

---

题干

给你一个由 n 个整数组成的数组 nums ，和一个目标值 target 。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]] （若两个四元组元素一一对应，则认为两个四元组重复）：

• 0 <= a, b, c, d < n
• a、b、c 和 d 互不相同
• nums[a] + nums[b] + nums[c] + nums[d] == target

你可以按 任意顺序 返回答案 。

示例 1：

输入：nums = [1,0,-1,0,-2,2], target = 0
输出：[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]

示例 2：

输入：nums = [2,2,2,2,2], target = 8
输出：[[2,2,2,2]]

---

解题思路

这道题和15.三数之和的思路是几乎一样的，思路是两层for循环来寻找a和b，把nums[i] + nums[j]看作整体，依然是循环内有left和right下标作为双指针，找出nums[i] + nums[j] + nums[left] + nums[right] == target的情况，三数之和的时间复杂度是O(n^2)，四数之和的时间复杂度是O(n^3). 但是这道题要注意的细节不少，主要集中在题干修改后的剪枝操作。与三数之和重复的细节我在这里不在罗列，详情请看三数之和这篇文章。

---

什么是剪枝？

剪枝是一种优化技术，用于提高算法的效率和性能，常应用于搜索算法、决策树和神经网络等领域。其基本概念如下：

在搜索算法中：剪枝指的是在搜索过程中提前排除那些不可能或不需要的分支，从而减少搜索空间，加快搜索速度。

---

运算符优先级

+- 号的优先级大于<>号大于&&

 完整代码如下：

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
      vector<vector<int>> result;
      sort(nums.begin(),nums.end());
      for(int i = 0; i < nums.size(); i++){
        //剪枝1   当数组存在负数时，再加一个数是可能导致和减小的，所以不能直接排除可能性
        if(nums[i] > target && nums[i] > 0){
            break;
        }
        //不要遗漏i>0，要确保不会溢出
        if(i > 0 && nums[i] == nums[i-1]){
            continue;
        }
        for(int j = i + 1; j < nums.size(); j++){
            //剪枝2 同样的操作，只不过把nums[i]+nums[j]看作是整体
            if(nums[i] + nums[j] > target && nums[i] + nums[j] > 0){
               break; 
            }
            不要遗漏j>i+1，要避免j-1=i
            if(j > i + 1 && nums[j] == nums[j -1]){
                continue;
            }
            int left = j + 1;
            int right = nums.size()-1;
            while(left < right){
                 //四个元素相加可能会溢出，所以用长整型
                 if((long)nums[i] + nums[j] + nums[left] + nums[right] > target){
                    right--;
                  }
                 else if((long)nums[i] + nums[j] + nums[left] + nums[right] < target){
                    left++;
                  }
                  else{
                    result.push_back({nums[i],nums[j],nums[left],nums[right]});
                    while(left < right && nums[left] == nums[left + 1]){
                       left++;
                    }
                    while(left < right && nums[right] == nums[right - 1]){
                       right--;
                   }
                    //前面是去重，还要更新指针的 
                    left++;
                   right--; 
                }
            }   
      }
     }
     return result;
    } 
};

改变数据类型格式：（long）+ 要改变的数据

当数组存在负数时，再加一个数是可能导致和减小的，所以只能在数组大于0的情况下剪枝