Subsets

  • 請編寫一個方法,返回某集合的全部非空子集。 給定一個int數組A和數組的大小int n,請返回A的全部非空子集。保證A的元素個數小於等於20,且元素互異。各子集內部從大到小排序,子集之間字典逆序排序,見樣例。 測試樣例:[123,456,789] 返回:{[789,456,123],[789,456],[789,123],[789],[456 123],[456],[123]}
import java.util.*;


public class Subset {
    public ArrayList<ArrayList<Integer>> getSubsets(int[] A, int n) {
        Integer [] integers = new Integer[A.length]; //我這裏寫的麻煩了一點, 實際上不用從大到小排序的,dfs的時候從後向前進行就能夠了
        for(int i = 0; i< A.length; ++i)
            integers[i] = A[i];
        Arrays.sort(integers, new MyComp());
        ArrayList<ArrayList<Integer>> res = new ArrayList<ArrayList<Integer>>();
        ArrayList<Integer> curr = new ArrayList<>();
        dfs(res, curr, 0, integers, n);
        res.remove(res.size() - 1);
        return res;
    }

    public void dfs(ArrayList<ArrayList<Integer>> res, ArrayList<Integer> curr, int index, Integer [] A, int n){
        for(int i = index; i < n; ++i){
            curr.add(A[i]);
            dfs(res, curr, i + 1, A, n);
            curr.remove(curr.size() - 1);
        }
        res.add((ArrayList<Integer>) curr.clone());
    }

    class MyComp implements Comparator<Integer>{
        @Override
        public int compare(Integer o1, Integer o2) {
            return o2 - o1;
        }
    }
}
相關文章
相關標籤/搜索