Leetcode - K Sum

Posted frankcui

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Leetcode - K Sum相关的知识,希望对你有一定的参考价值。

 

 

    List<List<Integer>> kSum_Trim(int[] a, int target, int k) 
        List<List<Integer>> result = new ArrayList<>();
        if (a == null || a.length < k || k < 2) return result;
        Arrays.sort(a);
        kSum_Trim(a, target, k, 0, result, new ArrayList<>());
        return result;
    

    void kSum_Trim(int[] a, int target, int k, int start, List<List<Integer>> result, List<Integer> path) 
        int max = a[a.length - 1];
        if (a[start] * k > target || max * k < target) return;

        if (k == 2)                         // 2 Sum
            int left = start;
            int right = a.length - 1;
            while (left < right) 
                if      (a[left] + a[right] < target) left++;
                else if (a[left] + a[right] > target) right--;
                else 
                    result.add(new ArrayList<>(path));
                    result.get(result.size() - 1).addAll(Arrays.asList(a[left], a[right]));
                    left++; right--;
                    while (left < right && a[left] == a[left - 1]) left++;
                    while (left < right && a[right] == a[right + 1]) right--;
                
            
        
        else                         // k Sum
            for (int i = start; i < a.length - k + 1; i++) 
                if (i > start && a[i] == a[i - 1]) continue;
                if (a[i] + max * (k - 1) < target) continue;
                if (a[i] * k > target) break;
                if (a[i] * k == target) 
                    if (a[i + k - 1] == a[i]) 
                        result.add(new ArrayList<>(path));
                        List<Integer> temp = new ArrayList<>();
                        for (int x = 0; x < k; x++) temp.add(a[i]);
                        result.get(result.size() - 1).addAll(temp);    // Add result immediately.
                    
                    break;
                
                path.add(a[i]);
                kSum_Trim(a, target - a[i], k - 1, i + 1, result, path);
                path.remove(path.size() - 1);        // Backtracking
            
        
    

 

以上是关于Leetcode - K Sum的主要内容,如果未能解决你的问题,请参考以下文章

[Leetcode]Top K问题总结

LeetCode 378. 有序矩阵中第K小的元素 | Python

leetcode笔记215. 数组中的第K个最大元素

[LeetCode]347. 前 K 个高频元素(堆)

[leetcode]347. Top K Frequent Elements

[LeetCode]215. 数组中的第K个最大元素(堆)