首页 > 代码库 > LeetCode 015 3Sum

LeetCode 015 3Sum

【题目】


Given an array S of n integers, are there elements abc 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;
    }
};