[LeetCode] 15.三数之和

[LeetCode] 15.三数之和

难度:Medium

题意:

[LeetCode] 15.三数之和_第1张图片

题解:

只用O(N*2)的复杂度,先进行排序。
固定i,确定left,right进行遍历。

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        if (nums.size() < 3)
        {
            return {};
        }

        vector<vector<int>> ans;

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

        for (int first = 0; first < nums.size(); first++)
        {
            if (nums[first] > 0)        // 已经有序,第一个数大于0则不可能和为0
            {
                return ans;
            }

            if (first > 0 && nums[first] == nums[first - 1])
            {
                continue;
            }
            
            int left = first + 1;
            int right = nums.size() - 1;
            while (left < right)
            {
                int temp = nums[first] + nums[left] + nums[right];
                if (temp == 0)
                {
                    ans.push_back({nums[first], nums[left], nums[right]});
                    while (left < right && nums[left] == nums[left+1])      // 跳出left相同的
                    {
                    left++;
                    }
                    while (left < right && nums[right] == nums[right-1])
                    {
                    right--;
                    }

                    left++;     // 移动到下一个不同的地方
                    right--;
                }else if (temp < 0)
                {
                    left++;     // 和小于0,左值右移
                }else
                {
                    right--;
                }   
            }
        }
        
        return ans;
    }
};

你可能感兴趣的:(LeetCode)