LeetCode 15.三数之和

Description:

题目大意：寻找三个不重复的数字，使其和为 0

解题思路:

算法标签：排序，双指针

1. 先进行排序，从小到大
2. 设置双指针 left 和 right
3. 以 nums[i] 为固定值，如果三个数之和大于0，右指针左移；如果三个数之和小于0，左指针右移
4. 注意去重

代码:

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        
        vector<vector<int>> ans;
        int len = nums.size();

        if(nums.size() < 3)
            return ans;

        sort(nums.begin(),nums.end());

        for(int i = 0;i < nums.size();i++) {
            // num[i] 如果大于0，其右边的值均大于 0，则三数之和不可能为 0
            // num[i] 为固定值
            if(nums[i] > 0)
                return ans;
            // 去除重复元素
            if(i > 0 && nums[i] == nums[i-1])
                continue;

            // 定义左右指针
            int left = i+1;
            int right  = len - 1;

            while(left < right) {
                // 如果三个数之和大于 0 ，右指针左移
                if(nums[i] + nums[left] + nums[right] > 0)
                    right--;
                // 如果三个数之和小于 0 ，右指针左移
                else if(nums[i] + nums[left] + nums[right] < 0)
                    left++;
                else {
                    ans.push_back({nums[i],nums[left],nums[right]});
                    // 去掉重复元素
                    while((left < right) && (nums[right] == nums[right-1]))
                        right--;
                    while((left < right) && (nums[left] == nums[left+1]))
                        left++;
                    // 左右指针同时移动，寻找新的结果
                    left++;
                    right--;
                }

            }
        }
        return ans;
    }
};