1. 程式人生 > >【LeetCode】4Sum 解題報告

【LeetCode】4Sum 解題報告

【題目】

Given an array S of n integers, are there elements abc, and d in S such that a + b + c + d = target? Find all unique quadruplets in the array which gives the sum of target.

Note:

  • Elements in a quadruplet (a,b,c,d) must be in non-descending order. (ie, a ≤ b ≤ c ≤ d)
  • The solution set must not contain duplicate quadruplets.
    For example, given array S = {1 0 -1 0 -2 2}, and target = 0.

    A solution set is:
    (-1,  0, 0, 1)
    (-2, -1, 1, 2)
    (-2,  0, 0, 2)
【解析】

3Sum 3Sum Closest 的擴充套件,同樣思路,加強理解。

K Sum 問題的時間複雜度好像為 O(n^(k-1)) ?!如果有更好的,歡迎指教!

【Java程式碼】

public class Solution {
    List<List<Integer>> ret = new ArrayList<List<Integer>>();
    
    public List<List<Integer>> fourSum(int[] num, int target) {
        if (num == null || num.length < 4) return ret;
        Arrays.sort(num);
        int len = num.length;
        for (int i = 0; i < len-3; i++) {
            if (i > 0 && num[i] == num[i-1]) continue;
            for (int j = i+1; j < len-2; j++) {
                if (j > i+1 && num[j] == num[j-1]) continue;
                findTwo(num, j+1, len-1, target, num[i], num[j]);
            }
        }
        return ret;
    }
    
    public void findTwo(int[] num, int begin, int end, int target, int a, int b) {
        if (begin < 0 || end >= num.length) return;
        int l = begin, r = end;
        while (l < r) {
            if (a+b+num[l]+num[r] < target) {
                l++;
            } else if (a+b+num[l]+num[r] > target) {
                r--;
            } else {
                List<Integer> ans = new ArrayList<Integer>();
                ans.add(a);
                ans.add(b);
                ans.add(num[l]);
                ans.add(num[r]);
                ret.add(ans);
                l++;
                r--;
                while (l < r && num[l] == num[l-1]) l++;
                while (l < r && num[r] == num[r+1]) r--;
            }
        }
    }
}