题目大意:给定一个整型数组,数组中允许存在重复元素,求出数组的所有子集。
理解: 1)与Subsets(http://blog.csdn.net/blog_szhao/article/details/24311891)要求相同,本题数组中存在重复的元素,相同元素交换位置后的集合是相同的;
2)由于要求所有子集以非递减排序,所以先对原数组排序;
3)统计数组中每个元素出现的次数,val[]数组记录对应下标元素出现的次数,同时将原数组只出现一次的元素合并在一起,以达到val[]的下标对应的值就是原数组对应下标元素出现的次数。
实现:
public class Solution {
private ArrayList<ArrayList<Integer>> res = new ArrayList<ArrayList<Integer>>();
private int len = 0;
private ArrayList<Integer> arr = new ArrayList<Integer>();
private int[] val;
public void init(int[] A) { // 统计出现次数
val = new int[len];
int i = 0, j = 1;
val[i] = 1;
while(j < len) {
if(A[j] != A[i]) {
if(j > i + 1) A[i + 1] = A[j]; // 合并非重复元素
i ++;
val[i] = 1;
}
else val[i] ++;
j ++;
}
}
public void func(int[] s, int pos) {
if(pos == len) {
ArrayList<Integer> tmp = new ArrayList<Integer>();
for(int i = 0; i < arr.size(); i ++) tmp.add(arr.get(i));
res.add(tmp);
return;
}
func(s, pos + 1);
for(int i = 0; i < val[pos]; i ++) {
arr.add(s[pos]);
func(s, pos + 1);
}
for(int i = 0; i < val[pos]; i ++) arr.remove(arr.size() - 1);
}
public ArrayList<ArrayList<Integer>> subsetsWithDup(int[] num) {
if(num == null || num.length == 0) return res;
len = num.length;
if(len == 1) {
arr.add(num[0]);
res.add(arr);
arr = new ArrayList<Integer>();
res.add(arr);
return res;
}
Arrays.sort(num);
init(num);
func(num, 0);
return res;
}
}