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 uniquetriplets 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 solutionset is:

    (-1, 0, 1)

    (-1, -1, 2)

解法：主要思想为选定第一个值num[i]，则转换成和为-num[i]的Two Sum问题（TwoSum的解法：

http://blog.csdn.net/woliuyunyicai/article/details/44222949）

在本题中要注意考虑消除重复结果的问题，通过跳过num[i+1]=num[i]的循环，以及在TwoSum子问题内部，消除重复现象（注意重复仅是指产生了多个相同的结果）

代码：

public class Solution{
	public ArrayList<ArrayList<Integer>> threeSum(int[] num) {
		// 结果数组
		ArrayList<ArrayList<Integer>> list = new ArrayList<ArrayList<Integer>>();
		//边界处理
		if(num.length < 3)
		{
			return list;
		}
		
		// 快速排序：或者直接使用Arrays.sort(num);
		quicksort(num, 0, num.length - 1);
		//Arrays.sort(num);

		// 先选定一个值,注意范围为0 ~ num.length - 2
		for (int i = 0; i < num.length - 2; i++) {
			// 消除重复值的重复计算，注意考虑 i = 0 情况
			if (i == 0 || num[i] > num[i - 1]  ) {
				int start = i + 1;
				int end = num.length - 1;
				int target = (-num[i]);// 转化为和为target的TwoSum问题
				while (start < end) {
					if (num[start] + num[end] > target) {
						--end;
					} else if (num[start] + num[end] < target) {
						++start;
					} else {
						ArrayList<Integer> temp = new ArrayList<Integer>();
						temp.add(num[i]);
						temp.add(num[start++]);
						temp.add(num[end--]);
						list.add(temp);
			
						//注意这里的判断条件，因为前面已经有了start++，end--（开始下一轮循环）
						//(这里要避免出现下面两个循环判断不成立，直接跳至下一循环的情况，若start，end不变化，则会始终循环下去)
						while (start < end && num[start] == num[start - 1]) {
							// 注意不可直接退出，应为可能还有其他两个数的和满足题设
							++start;
						}
						while (start < end && num[end] == num[end + 1]) {
							--end;
						}				
					}
				}
			}
		}
		return list;
	}

	// 先进行一次快速排序
	private void quicksort(int[] num, int start, int end) {
		if (start < end) {
			int mid = Partation(num, start, end);
			quicksort(num, start, mid - 1);
			quicksort(num, mid + 1, end);
		}
	}

	private int Partation(int[] num, int start, int end) {
		int midnum = num[end];// 选取哨兵
		int index = start;

		for (int i = start; i < end; i++) {
			if (num[i] <= midnum) {
				swap(num, index, i);
				++index;
			}
		}
		swap(num, index, end);
		return index;
	}

	// 交换两个位置上的数值
	private void swap(int[] data, int a, int b) {
		int temp = data[a];
		data[a] = data[b];
		data[b] = temp;
	}
}