leetcode -- 15. 3Sum 【问题转化2sum + 避免重复计算的方法（规定次序）】

题目

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: 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]
]

题意

在给定的数组S中寻找3个数的和为0的情况。

分析及解答

• 【转化】 将 3sum 转化为2sum的问题进行处理，即 固定一个 数，然后去按照2sum的方式进行处理。
• 【避免重复计算】为了避免一些不必要的元素重复加入到结果集中，我们需要适当的限定遍历的次序。（代码中标号3的注释部分）

代码

class Solution {
    public List> threeSum(int[] nums) {
     
        //1.排序。
    	Arrays.sort(nums);
    	//2.选取c（不同于上次所选）,然后 寻找a,b 满足 a + b = -c。
    	//3.a,b,c记录放入的集合S.(如何做到不重复放入呢？规定次序：只能在c的右边空间中寻找)
    	List> results = new ArrayList<>();
    	int c = 0;
    	for(int i = 0 ; i < nums.length ; i++){
    		if(i == 0){
    			c = nums[i];
    		}else if(nums[i] == c){
    			continue;
    		}else{
    			c = nums[i];
    		}
    		
    		for(int p1 = i+1,p2= nums.length -1 ;p1 < p2;){
    			if(nums[p1] + nums[p2] + c == 0){
    				//加入结果集中。
    				List result = new ArrayList<>(3);
    				result.add(c);
    				result.add(nums[p1]);
    				result.add(nums[p2]);
    				results.add(result);
    				while(p1 < p2 && nums[p1] == nums[++p1]){ // 这里出现了一些bug,注意。
    				}
    				while(p1 < p2 && nums[p2] == nums[--p2]){
    				}
    			}else if(nums[p1] + nums[p2] + c < 0){
    				p1++;
    				
    			}else if(nums[p1] + nums[p2] + c > 0){
    				p2--;
    			}
    		}
    	}
    	return results;
    }
}