登录
  • #刷题

3 ‌‌‌‌‍‍‌‍‌‌‍‍‌‍‌‌‌‍‌‌‍‍‌‍‍‌‌‌‌‍‌‌sum: 数字可重复使用的DFS解法复杂度是什么?

熟狗脸
1528
1
请教一下大家,对于数字可重复使用的3 sum,我写了下面的dfs解法,我感觉算法复杂度是 n*C(n, 3) 也就是说每次循环从n个数字中取最多3个数字的组合数,跑n次。但是我不太确定是否正确,大家觉得复杂度事多少呢?

对于输入[3, -3, 6, 2, 1]; 我程序的输出是: [[-3, -3, 6], [-3, 1, 2]]
    public static List<List<Integer>> threeSum2(int[] A) {[br][br][/br]//        Arrays.sort(A); // note: sorting not required any more here[br][/br][br][/br]        int n = A.length;[br][/br][br][/br]        List<List<Integer>> rst = new ArrayList<>();[br][/br][br][/br]        dfs2(A, 0, 0, 0, rst, new ArrayList<>());[br][/br][br][/br]        return rst;[br][/br][br][/br]    }[br][/br][br][/br]    private static void dfs2(int[] A, int index, int count, int sum, List<List<Integer>> rst, List<Integer> seg) {[br][br][/br]        if (count == 3 && sum == 0) {[br][/br][br][/br]            rst.add(new ArrayList<>(seg));[br][/br][br][/br]            return;[br][/br][br][/br]        }[br][/br][br][/br]        if ((count == 3 && sum != 0) || count > 3) return;[br][/br][br][/br]        for (int i = index; i < A.length; i++) {[br][/br][br][/br]            sum += A[i];[br][/br][br][/br]            seg.add(A[i]);[br][/br][br][/br]            dfs2(A, i, count + 1, sum, rst, seg);[br][/br][br][/br]            sum -= A[i];[br][/br][br][/br]            seg.remove(seg.size() - 1);[br][/br][br][/br]        }[br][/br][br][/br]    }[/i][/i][/i]
1条回复
热度排序

发表回复