LeetCode - 15. 3Sum

 

解题代码:

classSolution {

public:

    vector<vector<int>>threeSum(vector<int>& nums) {

        vector<vector<int>> res;

        if(nums.size()<=2)

            return res;

        sort(nums.begin(),nums.end());

        for(int i=0;i<nums.size();i++){

            int a=i+1,b=nums.size()-1;

            while(a<b){

               if(nums[a]+nums[b]+nums[i]>0)

                    b--;

                elseif(nums[a]+nums[b]+nums[i]<0)

                    a++;

                else{

                    vector<int> n(3,0);

                    n[0]=nums[i];

                    n[1]=nums[a];

                    n[2]=nums[b];

                    res.push_back(n);

                   while(a<b&&nums[b]==n[2])

                        b--;

                    while(a<b&&nums[a]==n[1])

                        a++;

                }

            }

           while(i<nums.size()-1&&nums[i+1]==nums[i])

                i++;

        }

        return res;

    }

};

 

解题思路:

题目要求数组中所有满足和为0的三个元素的组合。因此可以考虑使用双指针算法，先对数组进行排序，然后从第一个数开始，对于第i个数，第i+1和最后一个数分别定为a和b，由于数组为升序，因此，若第i、a、b个数之和小于0，则a+1,若大于0，则b-1，若等于0，则说明这是所要求的其中一组元素。其中需要注意的是，题目要求结果的元素里面没有重复，因此当我们遇到连续的相同数字时要跳过。