用比特串法求幂集的java实现

标签：

由于Java中有很方便的 String Integer.toBinaryString(int)，在学习生成子集的时后看到用比特串生成幂集的算法，就想着用java实现一下。幂集在解背包问题的时候还是很有用的，蛮力法，简单粗暴有效，当然仅限较小的实例。

实现如下 PowerSetGenerator.java，其中容器的实现用LinkedHashSet是为了保证集合中元素的位置固定，方便测试：

package powerset;

import java.util.Collection;
import java.util.LinkedHashSet;
import java.util.Iterator;

public class PowerSetGenerator<T> {
	public Collection<Collection<T>> getPowerSet(Collection<T> set){
		Collection<Collection<T>> r = new LinkedHashSet<Collection<T>>();
		int binary = 1<<set.size();
		for(int i=0;i<binary;i++){
			r.add(getSubSet(set,i));
		}
		return r;
	}
	
	private Collection<T> getSubSet(Collection<T> set,int binarySeq){
		Collection<T> sub = new LinkedHashSet<T>();
		int mask = 1 << set.size();
		Iterator<T> it = set.iterator();
		while(it.hasNext()){
			T obj = it.next();
			mask >>>= 1;
			if((binarySeq & mask)!=0){
				sub.add(obj);
			}
		}
		return sub;
	}
}

groovy单元测试 PowerSetTest.groovy，相当方便，特别是构建集合：

package powerset;

import static org.junit.Assert.*;

import org.junit.Test;

class PowerSetTest {

	static def set = [‘a‘,‘b‘,‘c‘]
	static def result = [[], [‘c‘], [‘b‘], [‘b‘, ‘c‘], [‘a‘], [‘a‘, ‘c‘], [‘a‘, ‘b‘], [‘a‘, ‘b‘, ‘c‘]]
	
	static def set2 = [1,2,3,4]
	static def result2 = [[], [4], [3], [3, 4], [2], [2, 4], [2, 3], [2, 3, 4], [1], [1, 4], [1, 3], [1, 3, 4], [1, 2], [1, 2, 4], [1, 2, 3], [1, 2, 3, 4]]
	
	static def set3 = [1]
	static def result3 = [[], [1]]
	@Test
	public void test() {
		PowerSetGenerator<Integer> generator = new PowerSetGenerator<Integer>();
		assertEquals(result.toString(),generator.getPowerSet(set).toString());
		assertEquals(result2.toString(),generator.getPowerSet(set2).toString());
		assertEquals(result3.toString(),generator.getPowerSet(set3).toString());
	}
}

用比特串法求幂集的java实现

标签：

原文地址：http://my.oschina.net/amhuman/blog/492390