Leetcode -- Combinations

来源:互联网 发布:网站特效源码 编辑:程序博客网 时间:2024/04/18 11:25

https://oj.leetcode.com/problems/combinations/

Given two integers n and k, return all possible combinations of k numbers out of 1 ... n.

For example,
If n = 4 and k = 2, a solution is:

[
  [2,4],
  [3,4],
  [2,3],
  [1,2],
  [1,3],
  [1,4],
]


public List<List<Integer>> combine(int n, int k)


和Combinations Sum 以及 Combinations Sum II是一类型的题目。建议大家也熟悉一下下一题Subset的做法,那是一切Combination类型题目的鼻祖,所有同类题目都是由那题延伸而来。另外还要注意区分一下Permutation和Subset的做法就好。这一题和最原始类型的combination的做法区别就在于这一题只需要且仅需要当长度为k的时候录入buf里的结果,其余都是不用的。

    public List<List<Integer>> combine(int n, int k) {        List<List<Integer>> res = new LinkedList<List<Integer>>();        if(k > n)            return res;        Integer[] buf = new Integer[k];        helper(res, buf, 0, 0, n, k);        return res;    }        public void helper(List<List<Integer>> res, Integer[] buf, int curpos, int curlevel, int n, int k){        if(curpos == k){            Integer[] cur_res = new Integer[k];//            System.arraycopy(buf, 0, cur_res, 0, k);            res.add(Arrays.asList(cur_res));//建议大家注意一下这里的写法,之所以要做拷贝再录入是因为asList这个函数会直接把当前数组作为list的内容            //也就是asList里面的实际数组内容就是buf这个reference。所以不能直接用buf        }else if(k - curpos > n - curlevel){            return;        }else{            for(int i = curlevel; i < n; i++){                buf[curpos] = i + 1;                helper(res, buf, curpos + 1, i + 1, n, k);            }        }    }


0 0
原创粉丝点击