查看: 2184| 回复: 1
跳转到指定楼层
上一主题 下一主题
收起左侧

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

全局:

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

您需要 登录 才可以下载或查看附件。没有帐号?注册账号

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.     }
复制代码

上一篇:刷题和刷项目只能二选一 时间(智商)不够真沮丧
下一篇:【Google】 面经, 这题怎么搞?
🔗
jingshihao 2017-7-31 09:56:39 | 只看该作者
全局:
复杂度感觉是N^3,每一次层递归有N种选择,最多递归三层。
其实跟3重for循环的解法是一个原理。
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 注册账号
隐私提醒:
  • ☑ 禁止发布广告,拉群,贴个人联系方式:找人请去🔗同学同事飞友,拉群请去🔗拉群结伴,广告请去🔗跳蚤市场,和 🔗租房广告|找室友
  • ☑ 论坛内容在发帖 30 分钟内可以编辑,过后则不能删帖。为防止被骚扰甚至人肉,不要公开留微信等联系方式,如有需求请以论坛私信方式发送。
  • ☑ 干货版块可免费使用 🔗超级匿名:面经(美国面经、中国面经、数科面经、PM面经),抖包袱(美国、中国)和录取汇报、定位选校版
  • ☑ 查阅全站 🔗各种匿名方法

本版积分规则

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