• Leetcode


        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
                }
            }
        }
  • 相关阅读:
    让电脑中的文件后缀显示完整
    [trie][异或] hdu 6625 three arrays
    [全排列] hdu 6628 permutation 1
    [模板][最大流]dinic
    [模板]矩阵十进制快速幂
    CCPC-Wannafly Summer Camp 2019 Day1
    [技巧]ARubbish
    [dp]第十届蓝桥杯国赛CB组C
    [暴力]分块
    [模板]主席树及其应用
  • 原文地址:https://www.cnblogs.com/frankcui/p/11546286.html
Copyright © 2020-2023  润新知