问题描述
|
我想找到所有通过回溯求和n的整数子集
例如,整数:
1 2 3 4 5 6 7
并且n = 7
我想输出
1 2 4
1 6
2 5
3 4
7
我认为我应该将我正在评估的整数数组中的位置作为参数传递,但是我仍然在写其余的逻辑。
到目前为止,我的代码:
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashSet;
import java.util.Set;
import java.util.TreeSet;
/**
*
* @author talleres
*/
public class Main {
int sum (TreeSet<Integer>ts,int temp) {
int sum=0;
for (Integer i: ts){
sum +=i;
}
return sum+temp;
}
static HashSet<TreeSet<Integer>> alternatives = new HashSet <TreeSet<Integer>>();
static ArrayList<TreeSet<Integer>> subsets = new ArrayList <TreeSet<Integer>>();
static TreeSet<Integer> getNextSubset (){
TreeSet<Integer> alternative = new TreeSet<Integer>();
if (!alternatives.contains(alternative)){
return alternative;
}
else return null; // BEWARE!!
}
static void findSubsets (ArrayList<Integer> numbers,int amount,int index){
TreeSet <Integer> subset = new TreeSet<Integer>();
int temp = numbers.get(index); //initialize alternative
if (temp<=amount)
subset.add(temp);
if (temp==amount)
subsets.add(subset);
}
public static void main(String[] args) throws IOException {
// Todo code application logic here
BufferedReader br = new BufferedReader(new InputStreamReader(system.in));
System.out.println(\"inset integers\");
ArrayList<Integer> numeros = new ArrayList<Integer>();
String line=br.readLine();
while (!line.equals(\"\")){
numeros.add (Integer.parseInt(line));
line = br.readLine();
}
Collections.sort(numeros);
System.out.println(\"insert the amount the subsets should sum\");
line = br.readLine();
int amount = Integer.parseInt(line);
ArrayList<Integer> accum = new ArrayList<Integer>();
findSubsets(numeros,amount,0);
}
}
解决方法
这是供您使用的一些伪代码:
Set<Set<Integer>> subsets(Set<Integer> remaining,int n) {
results = new HashSet<Set<Integer>>();
if (n == 0)
results.add(empty set);
for each i in remaining
newRemaining = remaining \\ {i}
for each subresult in subsets(newRemaining,n - i)
results.add(subresult + {i})
return results
}
对于负数也应该起作用。 (嗯,实际上可以用。在编写伪代码之前,我实现了它并对其进行了测试:-)
,我可能很想在递归函数中执行此操作。感觉很简单。可能不是最好的,但是效果很好。
这在伪代码中非常多,并假设数字为1..END。如果给定一个列表,则排序然后使用list [i]将是适当的。
find(int curpos,int cursum,int sumleft,char output[])
{
if (sumleft == 0)
print(output);
if (curpos > sumleft)
return;
for(i=curpos;i<=TARGET && i<=sumleft)
find(i+1,cursum+i,sumleft-i,output+i.\"+%d\")
}
main()
{
char output[100];
find(1,TARGET,\"\");
}