(LeetCode 15) 3Sum

来源:互联网 发布:手机淘宝店铺主图尺寸 编辑:程序博客网 时间:2024/05/12 22:48

Q:
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。并且输出时按照升序而且组合不能重复。

solution:
这道题思路和(LeetCode 16) 3Sum Closest是一样的。就不重复讲解了。其中有一点不同的就是,你要判断产生的组合是否有重复。
这道题的思路是:

  1. 先对给定的整型数组进行升序排序
  2. 从头开始遍历该数组,假设当前遍历到第i个数,暂时固定i
  3. 设置两个指针分别是i+1(j)一个指向数组尾(k),然后求这两个数组指针所指值与第i个数的和,判断这个和是不是当前target最近的。如果小于,那么j向后移动一位,否则k向前移动一位,直到j>=k
  4. i向前移动一位,重复第3步,直到整型数组全部被遍历

排序的时间复杂度是O(nlog(n))遍历的时间复杂度是O(n2),所以整个算法复杂度是O(n2)
这道题有几个可以剪枝的地方

  1. 若排序后的第一个元素都大于零或者最后一个数都小于零,那么就不存在这样的组合。
  2. 如果当前遍历的i已经大于零,那么i后面的数便可以不用再遍历
  3. 如果nums[i]+2*nums[j]>0那么就可以跳出循环,i向后移动一位进入新的循环。
class Solution {public:    vector<vector<int>> threeSum(vector<int>& nums) {        vector<vector<int>> res;        sort(nums.begin(),nums.end());        int size = nums.size();        if(size>0&&(nums[0]>0||nums[size-1]<0))return res;        for(int i=0;i<size;i++){            if(i>0&&nums[i]==nums[i-1])continue;            if(nums[i]>0)return res;            int j = i+1;            int k = size-1;            while(j<k){                if(nums[i]+2*nums[j]>0)break;                int sums = nums[i]+nums[j]+nums[k];                if(sums==0){                    vector<int> temp(3);                    temp[0]=nums[i];                    temp[1]=nums[j];                    temp[2]=nums[k];                    res.push_back(temp);                    j++;                    k--;                }                else if(sums<0)                  j++;                else k--;                while(j>i+1&&nums[j]==nums[j-1]){                    j++;                }                while(k<size-1&&nums[k]==nums[k+1]){                    k--;                }            }        }        return res;    }};
1 0
原创粉丝点击