LongDz 的数字文墨提案

15. 三数之和

3 min

题目

15. 三数之和 中等

思路

核心思想是排序后使用双指针技巧。首先固定一个数,然后在剩余部分用双指针寻找两数之和等于该数的相反数。

正确性基于数组有序性:当三数之和大于0时,右指针左移可减小总和;小于0时左指针右移可增加总和。跳过重复值确保不产生重复三元组,因为相同值的组合已被前置指针处理过。

  • Step 1: 对数组排序,便于双指针操作和去重
  • Step 2: 遍历数组,固定第一个数 nums[i],跳过重复值
  • Step 3: 初始化指针 j=i+1 和 k=n-1
  • Step 4: 当 j<k 时循环:若和大于0则 k—;若等于0则记录三元组并 j++
  • Step 5: 内层循环中跳过重复的 nums[j] 值

复杂度

  • 时间复杂度: O(n2)O(n^2),其中 nn 是数组长度。排序 O(nlogn)O(n \log n) 可忽略,双指针两层循环占主导。
  • 空间复杂度: O(logn)O(\log n),排序算法的栈空间开销(快速排序递归深度)。

代码

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        vector<vector<int>>ans;
        for(int i =0;i < nums.size();i++){
            if(i > 0 and nums[i] == nums[i-1])
            continue;
            int k = nums.size()-1;
            for(int j = i+1;j < nums.size();j++){
                if(j > i+1 and nums[j] == nums[j-1])
                continue;
                while(j < k and nums[i] + nums[j] + nums[k] > 0){
                    k--;
                }
                if(j == k)
                break;
                if(nums[i] + nums[j] + nums[k] == 0){
                    ans.push_back({nums[i],nums[j],nums[k]});
                }
            }
        }
        return ans;
    }
};