力扣三数之和、四数之和（数组+双指针+排序）

三数之和

算法流程

1、特判，对于数组长度 n，如果数组为 null 或者数组长度小于 3，返回 []。
2、对数组进行排序。
3、遍历排序后数组：

• 若 nums[i]>0：因为已经排序好，所以后面不可能有三个数加和等于 0，直接返回结果。
• 对于重复元素：跳过，避免出现重复解
• 令左指针 L=i+1，右指针R=n−1，当 L<R 时，执行循环：
  • 当 nums[i]+nums[L]+nums[R]==0，执行循环，判断左界和右界是否和下一位置重复，去除重复解。并同时将L,R 移到下一位置，寻找新的解
  • 若和大于 0，说明 nums[R]太大，R 左移
  • 若和小于 0，说明 nums[L] 太小，L 右移

转自 wu_yan_zu

代码及结果

class Solution {
    public static List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> ans = new ArrayList();
        int len = nums.length;
        if(nums == null || len < 3) return ans;
        Arrays.sort(nums); // 排序
        for (int i = 0; i < len ; i++) {
            if(nums[i] > 0) break; // 如果当前数字大于0，则三数之和一定大于0，所以结束循环
            if(i > 0 && nums[i] == nums[i-1]) continue; // 去重
            int L = i+1;
            int R = len-1;
            while(L < R){
                int sum = nums[i] + nums[L] + nums[R];
                if(sum == 0){
                    ans.add(Arrays.asList(nums[i],nums[L],nums[R]));
                    while (L<R && nums[L] == nums[L+1]) L++; // 去重
                    while (L<R && nums[R] == nums[R-1]) R--; // 去重
                    L++;
                    R--;
                }
                else if (sum < 0) L++;
                else if (sum > 0) R--;
            }
        }        
        return ans;
    }
}

转自 guanpengchn

图解

四数之和

代码及结果

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        List<List<Integer>> result = new ArrayList<>();
        if (nums == null || nums.length < 4) {
            return result;
        }
        Arrays.sort(nums);
        int length = nums.length;
        // 遍历
        for (int i = 0; i < nums.length - 3; i++) {
            // 剪枝
            if (nums[i] + nums[i + 1] + nums[i + 2] + nums[i + 3] > target) break;
            if (nums[i] + nums[length - 3] + nums[length - 2] + nums[length - 1] < target) continue;
            // 去重
            if (i > 0 && nums[i] == nums[i - 1]) continue;
            // 对第二个数进行循环
            for (int j = i + 1; j < nums.length - 2; j++) {
                // 剪枝
                if (nums[i] + nums[j] + nums[j + 1] + nums[j + 2] > target) break;
                if (nums[i] + nums[j] + nums[length - 2] + nums[length - 1] < target) continue;
                // 去重
                if (j > i + 1 && nums[j] == nums[j - 1]) continue;
                // 定义指针
                int left = j + 1, right = nums.length - 1;
                while (left < right) {
                    int sum = nums[i] + nums[j] + nums[left] + nums[right];
                    if (sum > target) {
                        right--;
                    } else if (sum < target) {
                        left++;
                    } else if (sum == target) {
                        result.add(Arrays.asList(nums[i], nums[j], nums[left], nums[right]));
                        // 在找到一个数组后进行去重
                        while (left < right && nums[right] == nums[right - 1]) right--;
                        while (left < right && nums[left] == nums[left + 1]) left++;
                        // 移动指针
                        right--;
                        left++;
                    }
                }                
            }
        }
        return result;
    }
}

注意事项

1.剪枝的优化，在三数之和中，我们使用的剪枝是 if(nums[i] > 0) return result; 但是在四数之和中，我们不能使用 if (nums[i] > target) break; ，因为 target 是一个不确定的数，对于 target < 0 的情况，这样是不成立的，因为排序后的数组可能前几个元素都是小于 0 但是大于 target 的。相加之后能够满足四数之和的要求。
2.循环中 i < nums.length - 3 而不是 i < nums.length 的处理技巧
转自 随机过程随机过