[英]Finding all the subsets that sum n via backtracking
我想找到所有通過回溯求和 n 的 integer 子集
例如對於整數:
1 2 3 4 5 6 7
n = 7
我想輸出
1 2 4
1 6
2 5
3 4
7
我認為我應該在我正在評估的 integer 數組中傳遞 position 作為參數,但我一直堅持編寫邏輯的 rest。
到目前為止我的代碼:
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
}
也應該適用於負數。 (嗯,實際上會工作。我在編寫偽代碼之前實現並測試了它:-)
我可能很想在遞歸 function 中執行此操作。 感覺很直接。 它可能不是最好的,但它會運作良好。
這在偽代碼中非常多,並假設數字是 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,0,TARGET,"");
}
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.