1. 程式人生 > 其它 >LeetCode-018-四數之和

LeetCode-018-四數之和

四數之和

題目描述:給定一個包含 n 個整數的陣列 nums 和一個目標值 target,判斷 nums 中是否存在四個元素 a,b,c 和 d ,使得 a + b + c + d 的值與 target 相等?找出所有滿足條件且不重複的四元組。

注意:答案中不可以包含重複的四元組。

示例說明請見LeetCode官網。

來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/4sum/
著作權歸領釦網路所有。商業轉載請聯絡官方授權,非商業轉載請註明出處。

解法一:雙指標法

首先,將nums排序;然後firstfourth指標分別從陣列的第一個和最後一位開始,second

third指標分別從first+1fourth-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();
        }
    }
}

【每日寄語】忠實的守住自己最初的夢想,讓生活的每一天都變得有意義。