1. 程式人生 > >LeetCode筆記——78子集

LeetCode筆記——78子集

題目:

給定一組不含重複元素的整數陣列 nums,返回該陣列所有可能的子集(冪集)。

說明:解集不能包含重複的子集。

示例:

輸入: nums = [1,2,3]
輸出:
[
  [3],
  [1],
  [2],
  [1,2,3],
  [1,3],
  [2,3],
  [1,2],
  []
]

思路:參考了大神們的程式碼和思路,原文連結:https://blog.csdn.net/happyaaaaaaaaaaa/article/details/51604217

程式碼採用了回溯的解法,回溯解法的基本格式是遞迴加迴圈。以[1,2,3]為例,開始時執行dfs(res,temp,nums,0),此時temp為空首先將空加入到結果集中,然後執行for迴圈,依次將1,2,3加入。每次都是將傳進來的emp先加入結果集在進行迴圈。加入的順序如下:[],[1],[1,2],[1,2,3],[1.3],[2],[2,3],[3].

程式碼:

public class Solution {
    public List<List<Integer>> subsets(int[] nums) {
        List<List<Integer>> res = new ArrayList<List<Integer>>();
        List<Integer> temp = new ArrayList<Integer>();
        dfs(res, temp, nums, 0);
        return res;
    }
    private void dfs(List<List<Integer>> res, List<Integer> temp, int[] nums, int j) {
        res.add(new ArrayList<Integer>(temp));
        for(int i = j; i < nums.length; i++) {
            temp.add(nums[i]);  //① 加入 nums[i]
            dfs(res, temp, nums, i + 1);  //② 遞迴
            temp.remove(temp.size() - 1);  //③ 移除 nums[i]
        }
    }
}

執行最快的程式碼:

這個程式碼和上面的思想不太一樣,總的來說是給結果集中的元素依次加入nums陣列中的元素。同樣以[1,2,3為例,]加入的順序為

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

class Solution {
    public List<List<Integer>> subsets(int[] nums) {
        List<List<Integer>> lists = new ArrayList<List<Integer>>();
		List<Integer> list = new ArrayList<Integer>();
		lists.add(list);
		return subsets(lists, nums, 0);
    }
    
    private List<List<Integer>> subsets(List<List<Integer>> lists, int[] nums, int n){
		if(n==nums.length) return lists;
		int length = lists.size();
		for(int i=0;i<length;i++) {
			List<Integer> list = lists.get(i);
			List<Integer> newlist = new ArrayList<Integer>();
			newlist.addAll(list);
			newlist.add(nums[n]);
			lists.add(newlist);
		}
		lists = subsets(lists, nums, n+1);
		return lists;
	}
}