Day13-代码随想录-三数之和15+四数之和18

三数之和题目如下所示：

这道题主要的难点就是怎么去重，但这个又是最重要的。用双指针的写法比哈希法更简单直观一些，主要思路如下面的代码所示：

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        //主要难在去重，不使用哈希法，使用双指针法
        List<List<Integer>> result = new ArrayList<>();
        //1.首先对数组进行排序
        Arrays.sort(nums);
        //2.对数组的每一个元素进行遍历：寻找a+b+c=0
        for(int i = 0; i < nums.length; i++){
            //3.判断特殊情况：如果排序后的第一个a就大于0，后面b+c绝对不可能小于0
            if (nums[i] > 0){
                return result;
            }
            //4.使用双指针法，定义left,right指针
            int left = i+1;
            int right = nums.length-1;
            //5.首先对a-nums[i]去重
            if (i >0 && nums[i] == nums[i-1]){
                continue;
            }
            //6.根据不同情况移动left和right指针
            while( right > left){
                int sum =nums[i] + nums[left] + nums[right];
                if (sum > 0){
                    right--;
                }else if(sum < 0){
                    left++;
                }else{//a+b+c=0的情况
                    result.add(Arrays.asList(nums[i],nums[left],nums[right]));
                    //在这个基础上对b和c去重
                    while(right > left && nums[right] == nums[right -1]) right--;
                    while(right > left && nums[left] == nums[left+1]) left++;
                    right--;
                    left++;
                }
            }
        }
        return result;
    }
}

四数之和题目如下所示（跟三数之和的解法差不多，只不过多了一层循环）

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        //方法与三数之和差不多，只不过多了一层循环
        List<List<Integer>> result = new ArrayList<>();
        Arrays.sort(nums);
        for(int k =0; k < nums.length; k++){
            if(nums[k] > target && target > 0){//这里不能单纯的只判断前者，因为taeget为负数的话就不行；
                break;
            }
            //对k进行去重操作
            if(k > 0 && nums[k] == nums[k-1]){
                continue;
            }
            //用双指针执行类似三数之和的操作
            for(int i = k + 1; i < nums.length; i++){
                //二级剪枝操作
                if(nums[k]+nums[i] > target && nums[k]+nums[i] > 0){
                    break;
                }
                //对i进行去重操作
                if(i > k+1 && nums[i] == nums[i-1]){
                    continue;
                }
                int left = i +1;
                int right = nums.length - 1;
                while(right > left){
                    long sum = (long) nums[k] + nums[i] + nums[left] + nums[right];
                    if(sum > target){
                        right--;
                    }else if(sum < target){
                        left++;
                    }else{
                        result.add(Arrays.asList(nums[k], nums[i], nums[left], nums[right]));
                        while(right > left && nums[right] == nums[right-1]) right--;
                        while(right > left && nums[left] == nums[left+1]) left++;
                        right--;
                        left++;
                    }
                }
            }
        }
        return result;
    }
}
/*也要学会写测试用例
public static void main(String[] args){
    Solution solution = new Solution();
    int[] nums = [1,0,-1,0,-2,2];
    int target = 0;
    List<List<Integer>> results = solution.fourSum(nums,target);
    for(List<Integer> result : results){
        System.out.println(result);
    }
}

*/