Leetcode18

思路与Leetcode15三数之和类似，外两层循环遍历两个数，再使用左右指针找另外两个数

数组必须先进行排序

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        List<List<Integer>> ans = new ArrayList<>();
        //特判
        if(nums.length < 4)
            return ans;

        int len = nums.length;
        Arrays.sort(nums);//排序很重要，不排序就没用了
        for (int i = 0; i < len - 3; i++) {
            //剪枝
            //排过序了，最小的四个数都比目标大，没必要再找了
            if(i + 3 < len && nums[i] + nums[i + 1] + nums[i + 2] + nums[i + 3] > target)
                break;
            //当前数和最大的三个数之和小于目标，可以跳过本次循环，没必要找了
            if(nums[i] + nums[len - 1] + nums[len - 2] + nums[len - 3] < target)
                continue;

            //去重
            if(i > 0 && nums[i] == nums[i - 1]){
                continue;
            }

            for (int j = i + 1; j < len - 2; j++) {
                //去重
                if(j > i + 1 && nums[j] == nums[j - 1]){
                    continue;
                }

                //剪枝
                //确定的两个数和最小的两个数之和都大于目标，没必要找了
                if(j + 2 < len && nums[i] + nums[j] + nums[j + 1] + nums[j + 2] > target)
                    break;
                //确定的两个数和最大的两个数之和小于目标，本次循环没必要找了
                if(nums[i] + nums[j] + nums[len - 1] + nums[len - 2] < target)
                    continue;

                int l = j + 1;
                int r = len - 1;
                while(l < r){
                    if(nums[i] + nums[j] + nums[l] + nums[r] == target){
                        ArrayList<Integer> list = new ArrayList<>();
                        list.add(nums[i]);
                        list.add(nums[j]);
                        list.add(nums[l]);
                        list.add(nums[r]);
                        ans.add(list);
                        //得到一种正确答案之后进行去重
                        l++;
                        while(l > j && l < len && nums[l] == nums[l - 1])
                            l++;
                        r--;
                        while(r < len - 1 && r > j && nums[r] == nums[r + 1])
                            r--;
                    } else if(nums[i] + nums[j] + nums[l] + nums[r] > target){
                        r--;
                    } else if (nums[i] + nums[j] + nums[l] + nums[r] < target){
                        l++;
                    }
                }
            }
        }
        return ans;
    }

    public static void main(String[] args) {
        int[] arr = {1,0,-1,0,-2,2};
        Solution solution = new Solution();
        List<List<Integer>> lists = solution.fourSum(arr,0);
        for (List<Integer> list : lists) {
            System.out.println(list);
        }
    }
}