leetcode-15:3sum 三数之和

题目:

Given an array nums of n integers, are there elements abc in nums such that ab + c = 0? Find all unique triplets in the array which gives the sum of zero.

Note:

The solution set must not contain duplicate triplets.

Example:

Given array nums = [-1, 0, 1, 2, -1, -4],

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

给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。

注意:答案中不可以包含重复的三元组。

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],

满足要求的三元组集合为:
[
  [-1, 0, 1],
  [-1, -1, 2]
]

思路:先对原数组排序,然后遍历排序后的数组。在循环内可以做如下优化:1.若当前遍历的 数是正数,就 break;因为当前数组已经是有序,后面的数字肯定是正数,和就不会是0,;2.加上重复跳过功能,,若第2个数与前面数相等则跳过。

我们使target=0-遍历过的数,然后查看剩下的元素是否为target。我们使用两个指针,i从当前遍历元素下一个开始从前往后,j从后往前,若两个数之和刚好为target,将当前元素,i和j一起放入ret,然后两个指针跳过重复元素。若两数之和小于target,则左指针向右移动一位,若大于target,j向左移动一位。

注意:1.若vector.size()需要用到两次以上,最好定义一个变量来保存vector的size(),这会省时间。

2.不要在循环体内定义变量,这也会造成浪费时间。

class Solution {
public:
    vector> threeSum(vector& nums) {
        vector> ret;
        sort(nums.begin(),nums.end());
        if(nums.empty()) return {};
        int len=nums.size(),i,j,target;
        for(int k=0;k0) break;
            if(k>0 && nums[k]==nums[k-1]) continue;
            i=k+1;
            j=len-1;
            target = 0-nums[k];
            while(i

参考:

http://www.cnblogs.com/grandyang/p/4481576.html

你可能感兴趣的:(数据结构/算法/刷题,#)