四数之和

2018-10-28  本文已影响0人  小白学编程

给定一个包含 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]
]

思路

最简单直接的暴力,不出意外地超时

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        List<List<Integer>> L = new ArrayList<List<Integer>>();
        Arrays.sort(nums);
        
        
        for (int i = 0; i < nums.length; i++) {
            
            for (int j = i + 1; j < nums.length; j++) {
                for (int k = j + 1; k < nums.length; k++) {
                    for (int l = k + 1; l < nums.length; l++) {
                        List<Integer> list = new ArrayList<Integer>();
                        if (nums[i] + nums[j] + nums[k] + nums[l] == target) {
                            list.add(nums[i]);
                            list.add(nums[j]);
                            list.add(nums[k]);
                            list.add(nums[l]);
                        }
                        int f = 1;
                        for (int a = 0; a < L.size(); a++) {
                            if(L.get(a).equals(list)) {
                                f = 0;
                            }
                        }
                        if (f == 1) {
                            if (list.size() != 0) {
                                 L.add(list);
                            }
                           
                        }
                        
                    }
                }
            }
        }
        return L;
        
    }
}

思路

先遍历两个数,剩下的两个数采用双指针的方式扫描

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        List<List<Integer>> L = new ArrayList<List<Integer>>();
        Arrays.sort(nums);
        for (int i = 0; i < nums.length; i++) {
            for (int j = i + 1; j < nums.length; j++) {
                int l = j + 1;
                int r = nums.length - 1;
                int temp = target - nums[i] - nums[j];
                while (l < r) {
                    if (nums[l] + nums[r] == temp) {
                        List<Integer> list = new ArrayList<>();
                        list.add(nums[i]);
                        list.add(nums[j]);
                        list.add(nums[l]);
                        list.add(nums[r]);
                        int f = 1;
                        for (int a = 0; a < L.size(); a++) {
                            if(L.get(a).equals(list)) {
                                f = 0;
                                break;
                            }
                        }
                        if (f == 1) {
                             L.add(list);
                        }
                        r--;
                        l++;
                    }else if (nums[l] + nums[r] > temp) {
                        r--;
                    }else if (nums[l] + nums[r] < temp) {
                        l++;
                    }
                    
                }
            }
        }
        return L;
        
    }
}
上一篇下一篇

猜你喜欢

热点阅读