15. 三数之和

题目

代码

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        //先固定住第一个数字，然后后面两个数字为 i+1和 nums.size()-1开始往中间缩小，并且要考虑数字重复的问题，时间复杂度为O(n²)
        vector<vector<int>> res;
        sort(nums.begin(),nums.end());
        
        for(int i=0;i<nums.size();i++)
        {
            int front=i+1;
            int back=nums.size()-1;
            int target=-nums[i];
            while(front<back)
            {
                int sum=nums[front]+nums[back];
                if(sum<target)
                {
                    front++;
                }else if(sum>target)
                {
                    back--;
                }else
                {
                    vector<int> temp;
                    temp.push_back(nums[i]);
                    temp.push_back(nums[front]);
                    temp.push_back(nums[back]);
                    res.push_back(temp);
                    
                    //移动到和当前的front指向的值不重复的地方
                    while(front<back&&nums[front]==temp[1]) front++;
                    //移动到和当前的back指向的值不重复的地方
                    while(front<back&&nums[back]==temp[2]) back--;

                }
                //移动到和当前i指向的值不重复的地方
                while(i+1<nums.size()&&nums[i+1]==nums[i]) i++;
                
            }
            
        }
        return res;
        
    }
};

思路

先将数组进行排序，比如[1,6,5,5,3]进行排序后为[1,3,5,5,6]，此时我们可以将问题化解为求两个数字的和等于第一个数字乘以-1.

假设第一个数字的取值为nums[i]，那么另外两个数字的取值从这个数组的nums[i+1]和nums.size()开始往中间取值，直到low<high不满足。