LeetCode 015 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.

Note:

• Elements in a triplet (a,b,c) must be in non-descending order. (ie, a ≤ b ≤ c)
• The solution set must not contain duplicate triplets.

    For example, given array S = {-1 0 1 2 -1 -4},

    A solution set is:
    (-1, 0, 1)
    (-1, -1, 2)

【题意】

给定一个数组，找出和为0的左右三数组合

两点注意：1. 不能有重复组合

2. 组合中的三个数要升序排列

【思路】

1. 对数组现升序排序

2. 维护三个指针p1,p2,p3分别用来确定组合中的第1,2,3个数

首先用p1指针从数组首元素开始依次确定第1个数。为了保证组合的唯一性，p1不能重复指向相同的数

在确定p1后，令p2=p1+1, p3=num.size()-1，这个时候就变成了"two sum"问题。

【注意，这里在做two sum的过程中需要注意p2的去重】

【代码】

class Solution {
public:
    vector<vector<int> > threeSum(vector<int> &num) {
        vector<vector<int> >result;
        int size=num.size();
        if(size<3)return result;                //判断数组长度是否大于3
        sort(num.begin(), num.end());           //排序
        for(int p1=0; p1<size-2; p1++){
            if(p1!=0 && num[p1]==num[p1-1])continue;    //第一位排重
            int p2=p1+1;
            int p3=size-1;
            while(p2<p3){
                if(p2!=p1+1 && num[p2]==num[p2-1]){p2++; continue;} //第二位排重
                int sum=num[p1]+num[p2]+num[p3];
                if(sum==0){
                    vector<int> triplet;
                    triplet.push_back(num[p1]);
                    triplet.push_back(num[p2]);
                    triplet.push_back(num[p3]);
                    result.push_back(triplet);
                    p2++;p3--;
                }
                else if(sum>0)p3--;
                else p2++;
            }
        }
        return result;
    }
};

LeetCode 015 3Sum,布布扣,bubuko.com