今天工作中遇到一個(gè)問(wèn)題需要一個(gè)組合數(shù)算法。google了一會(huì)，沒(méi)有找到合適的，于是就自己寫(xiě)了一個(gè)。主要是用了一個(gè)遞歸算法。發(fā)出來(lái)分享一下，有用的著的就拿去，覺(jué)得寫(xiě)得不好的就拍點(diǎn)磚，提點(diǎn)建議。
?
import java.util.*;
public class Combination
{
public static void main(String[] args)
{
Vector testData = new Vector(Arrays.asList(1, 2, 3, 4, 5, 6));
Vector results = getAllCombinations(testData);
for(int i=0; i<results.size(); i++)
System.out.println(results.elementAt(i));
}
public static Vector getAllCombinations(Vector data)
{
Vector allCombinations = new Vector();
for(int i=1; i<=data.size(); i++)
{
Vector initialCombination = new Vector();
allCombinations.addAll(getAllCombinations(data, i));
}
return allCombinations;
}
public static Vector getAllCombinations(Vector data, int length)
{
Vector allCombinations = new Vector();
Vector initialCombination = new Vector();
combination(allCombinations, data, initialCombination, length);
return allCombinations;
}
private static void combination(Vector allCombinations, Vector data,
Vector initialCombination, int length)
{
if(length == 1)
{
for(int i=0; i<data.size(); i++)
{
Vector newCombination = new Vector(initialCombination);
newCombination.add(data.elementAt(i));
allCombinations.add(newCombination);
}
}
if(length > 1)
{
for(int i=0; i<data.size(); i++)
{
Vector newCombination = new Vector(initialCombination);
newCombination.add(data.elementAt(i));
Vector newData = new Vector(data);
for(int j=0; j<=i; j++)
newData.remove(data.elementAt(j));
combination(allCombinations, newData, newCombination, length - 1);
}
}
}
}
?
測(cè)試結(jié)果：
[1]
[2]
[3]
[4]
[5]
[6]
[1, 2]
[1, 3]
[1, 4]
[1, 5]
[1, 6]
[2, 3]
[2, 4]
[2, 5]
[2, 6]
[3, 4]
[3, 5]
[3, 6]
[4, 5]
[4, 6]
[5, 6]
[1, 2, 3]
[1, 2, 4]
[1, 2, 5]
[1, 2, 6]
[1, 3, 4]
[1, 3, 5]
[1, 3, 6]
[1, 4, 5]
[1, 4, 6]
[1, 5, 6]
[2, 3, 4]
[2, 3, 5]
[2, 3, 6]
[2, 4, 5]
[2, 4, 6]
[2, 5, 6]
[3, 4, 5]
[3, 4, 6]
[3, 5, 6]
[4, 5, 6]
[1, 2, 3, 4]
[1, 2, 3, 5]
[1, 2, 3, 6]
[1, 2, 4, 5]
[1, 2, 4, 6]
[1, 2, 5, 6]
[1, 3, 4, 5]
[1, 3, 4, 6]
[1, 3, 5, 6]
[1, 4, 5, 6]
[2, 3, 4, 5]
[2, 3, 4, 6]
[2, 3, 5, 6]
[2, 4, 5, 6]
[3, 4, 5, 6]
[1, 2, 3, 4, 5]
[1, 2, 3, 4, 6]
[1, 2, 3, 5, 6]
[1, 2, 4, 5, 6]
[1, 3, 4, 5, 6]
[2, 3, 4, 5, 6]
[1, 2, 3, 4, 5, 6]
已有 3 人發(fā)表留言，猛擊->>這里<<-參與討論
JavaEye推薦