列表的可能组合

Question

我有一个对象数组列表，我想创建所有可能的组合（根据一组简单的规则）。存储在列表中的每个对象都包含一个小队编号和一个字符串。这是我存储的典型列表的示例：

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

我想要获取每个小队编号只能出现一次的所有组合，例如： (1,A),(2,A),(3,C),( 4,C) 那么下一个组合将是 (1,A),(2,A),(3,C),(4,D)。 我将如何在java中解决这个问题？通常我会使用嵌套循环，但事实上它全部存储在一个列表中，这让我的事情变得复杂。

谢谢， 脱漆剂

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);
}