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

二、解题思路

求解int数组中满足三数之和为0的组合。首先判断数组的长度是否大于2，如果不满足则返回空的列表；然后将数组中的元素按照从小到大的顺序排序，判断数组中的首元素和尾元素的乘积，如果积大于等于零，则返回空列表（这里有一个特殊情况就是三个元素都为0的情况），反之则开始寻找数组中满足条件的组合。定义三个位置变量，i（范围是从0到数nums.length-3）,j（范围是 从i+1到k）,k（范围是从nuns.length-1到i），固定变量I后，计算nums[i]+nums[j]+nums[k]的和，如果等于零，则存入列表中；如果大于零，则k--;如果小于零，则j++；

三、代码

public class Solution {    public List<List<Integer>> threeSum(int[] nums) {              List<List<Integer>> result=new ArrayList<List<Integer>>();        if(nums==null || nums.length<=2)            return result;        Arrays.sort(nums);        if(nums[0]*nums[nums.length-1]>=0 && nums[0]!=nums[nums.length-1])            return result;        //int middle=findMiddle(0,nums.length-1,nums);       // if(middle+1<nums.length-middle-1){            for(int i=0;i<nums.length-2;i++){                if(nums[i]>0)//如果num[i]>0，没有循环的必要了，因为三个数中的最小值已经大于零。                    break;                if(i==0 || (i>0 && nums[i]!=nums[i-1])){                    int j=i+1;                    int k=nums.length-1;                    while(j<k){                        if(nums[i]+nums[j]+nums[k]>0)                            k--;                        else if(nums[i]+nums[j]+nums[k]==0){                        //if(!result.contains(Arrays.asList(new Integer[]{nums[i],nums[j],nums[k]}))){                            result.add(Arrays.asList(new Integer[]{nums[i],nums[j],nums[k]}));                        //}                            while(j<k && nums[j]==nums[j+1]) j++;                            while(j<k && nums[k]==nums[k-1]) k--;                            j++;k--;                        }                                                else                            j++;                    }                }                            }              return result;            }    public int findMiddle(int start,int end,int[] nums){int middle=(end+start)/2;if((middle-start==1 && nums[start]*nums[middle]<=0) || middle==start)return start;else{if(nums[start]*nums[middle]>0)return findMiddle(middle, end,nums);elsereturn findMiddle(start, middle, nums);}}}

主要的问题是消除重复解，因此代码中会判断nums[i]和nums[i-1]，以及nums[j]和nums[j+1]，以及nums[k]和nums[k+1]是否相等，如果相等则跳过该元素。如果nums[i]的值大于零，即可结束循环，因为三个数中的最小值已经大于零。