LeetCoder 解题报告 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.

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，并输出这三个数，按从小到大输出，如果有多组就全部输出。

分析：在 LeetCode 解题报告 Two Sum 这篇文章中介绍了三种方法，此题是这个题的升级版，这个题如果用暴力解决那么将是O(n3)的时间复杂度，如果用HashMap也是不方便，因为数据是可重复的。

接下来就直接看代码

public List<List<Integer>> threeSum(int[] num) {
        Arrays.sort(num);
        List<List<Integer>> list = new ArrayList<List<Integer>> ();
        int first, end, mid;
        //遍历数组
        for(int i = 0; i < num.length-2; i++) {
        	if(i==0 || num[i] > num[i-1]) {
	        	first = i;
	        	end = num.length - 1;
	        	mid = first + 1;
	        	if(num[first] > 0 || num[end] < 0)
	        		break;
	        	while(mid < end) {
	        		int sum = num[first] + num[mid] + num[end];
	        		if(sum == 0) {
	        			ArrayList<Integer> each = new ArrayList<Integer>();
	        			each.add(num[first]);
	        			each.add(num[mid]);
	        			each.add(num[end]);
	        			if(!list.contains(each))
	        				list.add(each);
	        			mid++;
	        			end--;
	        			while(mid < end && num[end] == num[end-1])
	        				end--;
	        			while(mid < end && num[mid] == num[mid+1])
	        				mid ++;
	        		}
	        		else if(sum < 0) {
	        			mid++;
	        		}
	        		else 
	        			end--;
	        	}
        	}
        }
        return list;
    }

思路其实就很简单了，就是锁定第一个值遍历，剩下的两个的数，按照数组中寻找两个数的和是定值来解决就ok了。

如果按着思路写出的代码提交会有出现超时现象，那么我们就要优化了。

这里面做了很多的优化，比如：

if(i==0 || num[i] > num[i-1])

这里是防止相同数据输入

while(mid < end && num[end] == num[end-1])
	        				end--;
	        			while(mid < end && num[mid] == num[mid+1])
	        				mid ++;

这里避免不必要的重复