列表的可能组合

Question

我有一个对象的arraylist,我想创建所有可能的组合(根据一组简单的规则).存储在列表中的每个对象都包含一个squadNumber和一个字符串.以下是我存储的典型列表的示例:

0: 1, A
1: 1, B
2: 2, A
3: 2, B
4: 3, C
5: 3, D
6: 4, C
7: 4, D

我想获得每个squadNumber只能出现一次的所有组合,例如:(1,A),(2,A),(3,C),(4,C)然后下一个组合将是(1的A),(2,A),(3,C),(4,d).我将如何在java中解决这个问题？通常我会使用嵌套循环,但是它全部存储在一个列表中这一事实使我感到困惑.

谢谢,paintstripper

Answer 1

编辑

算法如下：

1. 按人数将所有小队分开。所以我们有 1 队的列表，还有 2 队的另一个列表，等等。
2. 运行 dfs. 在第 n 步，我们添加第 n 个号码的小队。

代码

// Split squads by numbers, so we can iterate through each number independently.
private Map<Integer, List<Squad>> splitSquadsByNumbers(List<Squad> squads) {
    Map<Integer, List<Squad>> res = new HashMap<Integer, List<Squad>>();
    for (Squad squad : squads) {
        if (res.get(squad.getNumber()) == null) {
            res.put(squad.getNumber(), new ArrayList<Squad>());
        }
        res.get(squad.getNumber()).add(squad);
    }
    return res;
}

List<Integer> squadNumbers;
Map<Integer, List<Squad>> squadsByNumbers;
Stack<Squad> stack;

// Iterating through each squad with number squadNumbers[position] and try to add to stack, at the end pop it from stack.

private void dfs(int position) {
    if (position == squadNumber.size()) {
        System.out.println(stack.toString());
    } else {
        for (Squad squad : squadsByNumbers.get(squadNumber.get(position))) {
            stack.push(squad);
            dfs(position + 1);
            stack.pop();
        }
    }
}

private void main(List<Squad> squads) {
    squadsByNumbers = splitSquadsByNumbers(squads);
    squadNumber = new ArrayList(squadsByNumber.keySet());
    Collections.sort(squadNumbers);
    stack = new Stack<Squad>();
    dfs(0);
}