c++实现三数之和

题目：

给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请

你返回所有和为 0 且不重复的三元组。

注意：答案中不可以包含重复的三元组

这个三数之和的算法实现步骤如下:1. 先对数组 nums 进行排序。这个步骤是为了让后续的查找更高效,通过排好序的数组可以使用左右指针进行二分查找。

2. 以 nums 数组的第一个元素 nums[i] 作为基准,使用左右指针查找 nums[i] 的两侧是否存在满足 nums[i] + nums[left] + nums[right] == 0 的两个元素。

3. 左指针 left 初始化为 i + 1,右指针 right 初始化为数组的末尾。

4. 如果 nums[i] + nums[left] + nums[right] == 0,说明找到了一个三元组解。将其插入 set,并移动左右指针。

5. 如果和大于 0,说明 nums[right] 太大,right 左移。如果和小于 0,说明 nums[left] 太小,left 右移。

6. 重复步骤 4 和 5,直到左右指针交叉。

7. 重复步骤 2 到 6,以后续元素为基准继续查找,直到处理完数组所有元素。

8. 将 set 中存储的所有三元组解,存入 vector 并返回。

举个例子,如果 nums = [-1, 0, 1, 2, -1, -4],这个算法的执行过程是:

1. 排序后 nums = [-4, -1, -1, 0, 1, 2]

2. 以 nums[0] = -4 为基准,left = 1,right = 5。-4 + -1 + 2 != 0,left 右移

3. left = 2,right = 5。-4 + -1 + 1 == 0,找到一组解,left 和 right 继续移动

4. left = 3,right = 4。-4 + 0 + 2 != 0,right 左移 

5. 重复步骤 3 和 4,找到第二组解 -4 + -1 + 1,将两个解插入 set

6. 以后续元素为基准继续查找,找到第三组解 -1 + -1 + 27. 返回结果 { {-4, -1, 1}, {-4, -1, 1}, {-1, -1, 2} }

cpp
class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        // 使用 set 存储不重复的三元组
        set<vector<int>> tem;  
        // 先对数组排序,方便后续查找
        sort(nums.begin(),nums.end());  
        // 数组长度
        int len=nums.size();   
        // 左右指针
        int left,right;   
        // 以第一个数 nums[i] 进行查找,找满足 nums[i] + nums[left] + nums[right] == 0 的组合
        for(int i=0;i<len;i++){   
            // 左指针从第二个数字开始
            left=i+1;  
            // 右指针从最后一个数字开始
            right=len-1;  
            // 当左右指针相遇时结束循环
            while(left<right){   
                // 如果找到了和为 0 的三元组
                if(nums[i]+nums[left]+nums[right]==0){   
                    // 将三元组插入到 set 中  
                    tem.insert({nums[i],nums[left],nums[right]});    
                    // 左指针右移
                    left++;    
                    // 右指针左移 
                    right--;
                }
                else if(nums[i]+nums[left]+nums[right]<0){ // 如果三元组之和小于 0   
                    // 左指针右移
                    left++; 
                }
                else{   // 如果三元组之和大于 0
                    // 右指针左移
                    right--; 
                }
            }
        }
        // 将 set 中的三元组存入结果集合
        vector<vector<int>> ans;
        for(auto &x:tem){   
            ans.push_back(x);
        }
        return ans;
    }
};