《数据科学面试40+真题讲解》,K神本年度最后一次开课


一亩三分地论坛

 找回密码
 Sign Up 注册获取更多干货
码农求职神器Triplebyte:
不用海投,内推你去多家公司面试
Airbnb 数据科学职位
in analytics and inference
天天打游戏、照样领工资,
你要不要来?
把贵司招聘信息放这里
查看: 326|回复: 1
收起左侧

[算法题] 3 sum: 数字可重复使用的DFS解法复杂度是什么?

[复制链接] |试试Instant~ |关注本帖
熟狗脸 发表于 2017-7-30 10:07:02 | 显示全部楼层 |阅读模式

注册一亩三分地论坛,查看更多干货!

您需要 登录 才可以下载或查看,没有帐号?Sign Up 注册获取更多干货

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

对于输入[3, -3, 6, 2, 1]; 我程序的输出是: [[-3, -3, 6], [-3, 1, 2]]
  1.     public static List<List<Integer>> threeSum2(int[] A) {
  2. //        Arrays.sort(A); // note: sorting not required any more here
  3.         int n = A.length;
  4.         List<List<Integer>> rst = new ArrayList<>();

  5.         dfs2(A, 0, 0, 0, rst, new ArrayList<>());

  6.         return rst;
  7.     }

  8.     private static void dfs2(int[] A, int index, int count, int sum, List<List<Integer>> rst, List<Integer> seg) {
  9.         if (count == 3 && sum == 0) {
  10.             rst.add(new ArrayList<>(seg));
  11.             return;
  12.         }
  13.         if ((count == 3 && sum != 0) || count > 3) return;

  14.         for (int i = index; i < A.length; i++) {
  15.             sum += A[i];
  16.             seg.add(A[i]);
  17.             dfs2(A, i, count + 1, sum, rst, seg);
  18.             sum -= A[i];
  19.             seg.remove(seg.size() - 1);
  20.         }
  21.     }
复制代码
jingshihao 发表于 2017-7-31 09:56:39 | 显示全部楼层
复杂度感觉是N^3,每一次层递归有N种选择,最多递归三层。
其实跟3重for循环的解法是一个原理。
回复 支持 反对

使用道具 举报

本版积分规则

关闭

一亩三分地推荐上一条 /5 下一条

手机版|小黑屋|一亩三分地论坛声明

custom counter

GMT+8, 2017-11-22 15:35

Powered by Discuz! X3

© 2001-2013 Comsenz Inc. Design By HUXTeam

快速回复 返回顶部 返回列表