四数之和
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;
}
}