LeetCode-018-四數之和
阿新 • • 發佈:2021-06-23
四數之和
題目描述:給定一個包含 n 個整數的陣列 nums 和一個目標值 target,判斷 nums 中是否存在四個元素 a,b,c 和 d ,使得 a + b + c + d 的值與 target 相等?找出所有滿足條件且不重複的四元組。
注意:答案中不可以包含重複的四元組。
示例說明請見LeetCode官網。
來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/4sum/
著作權歸領釦網路所有。商業轉載請聯絡官方授權,非商業轉載請註明出處。
解法一:雙指標法
首先,將nums排序;然後first和fourth指標分別從陣列的第一個和最後一位開始,second
和third指標分別從first+1和fourth-1處從兩邊向內移動,直到second不小於third,移動過程中需要判斷4個指標所指向的數字之和是否和target相等,如果相等,則放到結果集result裡面。 直到遍歷到first不小於fourth-2為止,最後返回結果result。
import java.util.ArrayList; import java.util.Arrays; import java.util.List; public class Solution { public static List<List<Integer>> fourSum(int[] nums, int target) { if (nums == null || nums.length < 4) { return new ArrayList<>(); } if (nums.length == 4 && nums[0] + nums[1] + nums[2] + nums[3] != target) { return new ArrayList<>(); } Arrays.sort(nums); List<List<Integer>> result = new ArrayList<>(); for (int first = 0, fourth = nums.length - 1; first < fourth - 2; first++) { // 過濾掉重複的 if (first > 0 && nums[first] == nums[first - 1]) { continue; } while (first < fourth - 2) { // 過濾掉重複的 if (fourth < nums.length - 1 && nums[fourth] == nums[fourth + 1]) { fourth--; continue; } for (int second = first + 1, third = fourth - 1; second < third; second++) { // 過濾掉重複的 if (second > first + 1 && nums[second] == nums[second - 1]) { continue; } while (second < third && nums[first] + nums[second] + nums[third] + nums[fourth] > target) { third--; } if (second != third && nums[first] + nums[second] + nums[third] + nums[fourth] == target) { result.add(new ArrayList<>(Arrays.asList(new Integer[]{nums[first], nums[second], nums[third], nums[fourth]}))); } } fourth--; } fourth = nums.length - 1; } return result; } public static void main(String[] args) { int[] nums = new int[]{-3, -1, 0, 2, 4, 5}; for (List<Integer> integers : fourSum(nums, 0)) { for (Integer integer : integers) { System.out.print(integer + "/"); } System.out.println(); } } }
【每日寄語】忠實的守住自己最初的夢想,讓生活的每一天都變得有意義。