leetcode78. Subsets

weixin_33840661發表於2017-07-08

題目要求

Given a set of distinct integers, nums, return all possible subsets.

Note: The solution set must not contain duplicate subsets.

For example,
If nums = [1,2,3], a solution is:

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

類似的題目有:
leetcode60 Permutation Sequence 可以參考這篇部落格
leetcode77 Combinations 可以參考這篇部落格

思路一:遞迴

還是利用遞迴的方式,在前一種情況的基礎上遍歷下一輪的組合情況。

    public List<List<Integer>> subsets(int[] nums) {
        List<List<Integer>> result = new ArrayList<List<Integer>>();
        result.add(new ArrayList<Integer>());
        subsets(result, nums, 0, new ArrayList<Integer>());
        return result;
    }
    
    public void subsets(List<List<Integer>> result, int[] nums, int startIndex, List<Integer> currentList){
        if(startIndex == nums.length){
            return;
        }
        while(startIndex<nums.length){
            currentList.add(nums[startIndex++]);
            result.add(new ArrayList<Integer>(currentList));
            subsets(result, nums, startIndex, currentList);
            currentList.remove(currentList.size()-1);
        }
    }

思路2:排序後迴圈

起始subset集為:[]
新增S0後為:[], [S0]
新增S1後為:[], [S0], [S1], [S0, S1]
新增S2後為:[], [S0], [S1], [S0, S1], [S2], [S0, S2], [S1, S2], [S0, S1, S2]

public List<List<Integer>> subsets(int[] S) {
    List<List<Integer>> res = new ArrayList<>();
    res.add(new ArrayList<Integer>());
    
    for(int i : S) {
        List<List<Integer>> tmp = new ArrayList<>();
        for(List<Integer> sub : res) {
            List<Integer> a = new ArrayList<>(sub);
            a.add(i);
            tmp.add(a);
        }
        res.addAll(tmp);
    }
    return res;
}

clipboard.png
想要了解更多開發技術,面試教程以及網際網路公司內推,歡迎關注我的微信公眾號!將會不定期的發放福利哦~