[LeetCode]18-四数之和(Medium)

题目

给定一个包含n个整数的数组nums和一个目标值target,判断nums中是否存在四个元素 a,b,c,d 使得a+b+c+d的值与target相等？
找出所有满足条件且不重复的四元组。

示例

给定数组 nums = [1, 0, -1, 0, -2, 2]，和 target = 0。

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

解题思路

• 当时做题18的时候，先是做了1-两数之和，15-三树之和，454-四数相加Ⅱ三道题。从两数之和到三数之和当时用的思想就是化简，将三数之和的问题化简到两数之和问题，然后套用经典的双指针就可以解决了。同理本题的四数之和同样可以沿着化简的思路，一路化简到三数之和，两数之和。
• 下面的话，直接用代码实现一下，主要要注意的地方就是边界情况以及如何去避免冗余计算(考虑这一点可以很大的优化时间复杂度)。代码是可以AC的。
  

代码

public List<List<Integer>> fourSum(int[] nums, int target) {
	List<List<Integer>> ret = new ArrayList<>();
	//特殊情况的检查
	if(nums==null||nums.length<4){
    	return ret;
    }
	Arrays.sort(nums);
	int len = nums.length;
	//i,j,start,end分别是四个指针
	for(int i = 0; i < len - 3 ; i++){
		//去重,如果当前元素和上一个元素相等的话，因为上一个元素已经考虑过了，所以当前元素就不需要考虑了。
		if( i > 0 && nums[i] == nums[i-1]){
			continue;
		}
		
		//避免冗余计算，主要是在两种情况下可以不继续往后计算了。
		//1：当前元素的最小四数之和如果都大于target的话，显然后面不可能有等于target的组合，且当前元素已经不能再变大，直接退出循环。
		//2：当前元素的最大四数之和如果都小于target的话，当前元素也就不会存在满足的组合，当前元素需要后移一位。
		int minValue = nums[i] + nums[i+1] + nums[i+2] + nums[i+3];
		if(minValue > target)
			break;
		int maxValue = nums[i] + nums[len-3] + nums[len-2] + nums[len - 1];
		if(maxValue < target)
			continue;//此时实现了i++，即右移。
		
		//上面一重for循环已经将四数之和化简到了三数之和，现在化简到两数之和。
		for(int j = i + 1; j < len -2 ; j++){
			//同理，去重
			if(j > i+1 && nums[j] == nums[j-1])
				continue;
			
			//继续避免冗余计算，思路同上
			int minValue2 = nums[i] + nums[j] + nums[j+1] + nums[j+2];
			int maxValue2 = nums[i] + nums[j] + nums[len-2] + nums[len-1];
			if(minValue2 > target)
				break;//如果此时j和i之间有间隔的话，i->i+1，j->i+2，这种情况下可能还有解
			if(maxValue2 < target)
				continue;//j++,右移
			//此时就已经化简到了两数之和的部分了
			int start = j + 1, end = len -1;
			while(start < end){
				int curValue = nums[i] + nums[j] + nums[start] + nums[end];
				if(curValue == target){
					ret.add(Arrays.asList(nums[i],nums[j],nums[start],nums[end]));
					//维护start和end指针，先是去重(找到和当前元素不一样的下个元素)然后以动
					while(start < end && nums[start] == nums[start+1]) start++;
					while(start < end && nums[end] == nums[end-1]) end--;
					start++;
					end--;
				}
				else if( curValue > target)
					end--;
				else
					start++;
			}
		}
	} 
	return ret;
}