注册一亩三分地论坛,查看更多干货!
您需要 登录 才可以下载或查看附件。没有帐号?注册账号
x
请教一下大家,对于数字可重复使用的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) {
- // Arrays.sort(A); // note: sorting not required any more here
- int n = A.length;
- List<List<Integer>> rst = new ArrayList<>();
- dfs2(A, 0, 0, 0, rst, new ArrayList<>());
- return rst;
- }
- private static void dfs2(int[] A, int index, int count, int sum, List<List<Integer>> rst, List<Integer> seg) {
- if (count == 3 && sum == 0) {
- rst.add(new ArrayList<>(seg));
- return;
- }
- if ((count == 3 && sum != 0) || count > 3) return;
- for (int i = index; i < A.length; i++) {
- sum += A[i];
- seg.add(A[i]);
- dfs2(A, i, count + 1, sum, rst, seg);
- sum -= A[i];
- seg.remove(seg.size() - 1);
- }
- }
复制代码 |