LeetCode15 三数之和

前言

题目： 15. 三数之和
文档： 代码随想录——三数之和
编程语言： C++
解题状态： 没思路…

思路

不可包含重复三元组的条件是本题最大的难点，本题的一大思路在与排序后进行去重。

代码

双指针法

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> result;
        sort(nums.begin(), nums.end());
        for (int i = 0; i < nums.size(); i++) {
            if (nums[i] > 0) return result;

            if (i > 0 && nums[i] == nums[i - 1]) continue;

            int left = i + 1;
            int right = nums.size() - 1;

            while (right > left) {
                if (nums[i] + nums[left] + nums[right] > 0) {
                    right--;
                } else if (nums[i] + nums[left] + nums[right] < 0) {
                    left++;
                } else {
                    result.push_back({nums[i], nums[left], nums[right]});
                    while (right > left && nums[right] == nums[right - 1]) right--;

                    while (right > left && nums[left] == nums[left + 1]) left++;

                    right--;
                    left++;
                }
            }
        }

        return result;
    }
};

• 时间复杂度： $O(n^2)$
• 空间复杂度： $O(1)$