3Sum

Given an array S of n integers, are there elements a, b, c in S such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.
For example, given array S = [-1, 0, 1, 2, -1, -4],

A solution set is:
[
[-1, 0, 1],
[-1, -1, 2]
]

• 题目为所求数组中三项相加和为零的所有情况。
• 最简单的想法，三个for循环遍历，结果超时毫无疑问。
• 我们可以类似模仿上一次博客的题目，先排序，从两边往中间遍历。

思路

首先选择一个基础遍历项i，另外两项之和满足相加为-i即可

• 从第0项到第n-2项遍历
• 左边（J）从i+1项开始，右边（k）从n项开始
• 确保J小于k，如果j+k>-i，则k–；如果j+k<-i，则j++。

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        if (nums.size() <= 2)
            return{};
        vector<vector<int>> res;
        sort(nums.begin(), nums.end());
        for (int i = 0; i < nums.size() - 2; i++) 
        {
            if (i == 0 ||(i>0&& nums[i] != nums[i-1])) 
            {

                int j = i + 1, k = nums.size() - 1;
                int tag = 0 - nums[i];
                while (j < k) 
                {

                    if (nums[j] + nums[k] == tag) 
                    {
                        res.push_back({ nums[i],nums[j],nums[k] });
                        while (j < k&&nums[j] == nums[j + 1])
                            j++;
                        while (j < k&&nums[k] == nums[k - 1])
                            k++;
                        j++;
                        k--;

                   }
                    else if (nums[j] + nums[k] > tag) k--;
                    else j++;

                }
            }

        }

        return res;
    }
};