Java 生成所有子集

问题描述:给定一个数组A和数组的大小n,返回A的所有子集。保证A的元素互异。

解法1:可以将问题分解,先求出只有1个元素的子集,然后求只有两个元素的子集,一直求到n;然后进行统计时加上空集。这样就把问题分解成n个组合问题。但这样效率比较低。这里省略不讲。

解法2:采用逐步生成的办法,先假设只求一个元素的所有子集,容易得出结果为空集和元素本身。然后再这个基础上去求2个元素的所有子集;由于现在多了一个元素,所以原有的所有子集就面临着一个问题:添加这个元素和不添加这个元素。也就是说多一个元素后所有的子集就等于(原有的所有子集)+(原有的每一个子集添加这个元素后的结果)。

代码如下:


import java.util.*;

public class 生成子集 {
	public static void main(String[] args){
		int[] arr = {1,2,3};
		System.out.println(getSubsets(arr, arr.length, arr.length-1));
	}

	private static Set<Set<Integer>> getSubsets(int[] arr, int n, int cur) {
		Set<Set<Integer>> newset = new HashSet();
		if(cur==0){//处理第一个元素,
			Set<Integer> nul = new HashSet();//空集
			Set<Integer> fil = new HashSet();//包含第一个元素的集合
			fil.add(arr[cur]);
			newset.add(nul);
			newset.add(fil);
			return newset;
		}
		Set<Set<Integer>> oldset = getSubsets(arr, n, cur-1 );
		for(Set<Integer> set:oldset){
			newset.add(set);//保留原样,不添加新元素
			//由于直接给set添加元素会导致被破坏,所以要克隆set
			//然后用克隆后的set进行添加
			//由于set只是接口,无法调用clone()函数,
			//所以要转换为HashSet再调用clone()函数
			Set<Integer> clone = (Set<Integer>)((HashSet)set).clone();
			clone.add(arr[cur]);//子集添加新元素
			newset.add(clone);//把添加新元素后的子集添加进来		
		}
		return newset;
	}
}

解法3利用二进制求解,将数组中的n个元素对应n位的二进制。其中每一个元素对应一个二进制位。那么从0到2的n次方减一的每一个数就可以对应一个子集。也就是说所有子集一共有2的n次方个。
  这个解法的好处是更简洁,而且能将所有子集按照字典序输出。

代码如下:


import java.util.*;

public class 生成子集二进制解法 {
	public static void main(String[] args){
		int[] arr = {1,3,2};
		System.out.println(getSubsets(arr, arr.length));
	}

	private static ArrayList<ArrayList<Integer>> getSubsets(int[] arr, int n) {
		Arrays.sort(arr);//正序排序,保证按照字典顺序排序
		ArrayList<ArrayList<Integer>> set = new ArrayList();
		for(int i=f(n)-1; i>=0; i--){
			ArrayList a = new ArrayList();
			for(int j=n-1; j>=0; j--){
				if(((i >> j)&1)==1){//判断j位上是否为1,从高位开始检查
					a.add(arr[j]);
				}
			}
			set.add(a);
		}
		return set;
	}

	private static int f(int n) {
		int sum = 1;
		for(int i=0; i<n; i++){
			sum = sum*2;
		}
		return sum;
	}
}
发布了45 篇原创文章 · 获赞 5 · 访问量 1934

猜你喜欢

转载自blog.csdn.net/HC199854/article/details/104455133