请编写一个方法,返回某集合的所有非空子集。
给定一个int数组A和数组的大小int n,请返回A的所有非空子集。保证A的元素个数小于等于20,且元素互异。各子集内部从大到小排序,子集之间字典逆序排序,见样例。
测试样例:
[123,456,789]
返回:{[789,456,123],[789,456],[789,123],[789],[456 123],[456],[123]}
解析:典型的DFS路径问题。根据题意,最好先逆序排列后再DFS,用一个临时变量存放遍历到的路径,add1次遍历后remove1次再遍历就能得到倒序的排列。
Tips:最后使用临时变量添加入返回列表的时候需要注意数组及对象是引用传址的问题,所以需要new出一个新的变量再添加。
代码:
private ArrayList<ArrayList<Integer>> list = new ArrayList<>();
private ArrayList<Integer> _A = new ArrayList<>();
private ArrayList<Integer> temp = new ArrayList<>();
private void dfs(int index) {
if (index > _A.size() - 1) {
if (!temp.isEmpty() && !list.contains(temp)) {
list.add(new ArrayList<Integer>(temp));
}
return;
}
temp.add(_A.get(index));
dfs(index + 1);
temp.remove(temp.size() - 1);
dfs(index + 1);
}
public ArrayList<ArrayList<Integer>> getSubsets(int[] A, int n) {
for (int i = 0; i < A.length; ++i) {
_A.add(A[i]);
}
_A.sort(new Comparator<Integer>() {
@Override
public int compare(Integer o1, Integer o2) {
return o2.compareTo(o1);
}
});
dfs(0);
return list;
}