组合问题-含有重复元素（回溯算法）

package org.example;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class Main {

    // 主方法，用于测试
    public static void main(String[] args) {
        int[] nums = {2, 2, 3, 3};
        int k = 3;
        List<List<Integer>> result = combine(nums, k);
        for (List<Integer> combination : result) {
            System.out.println(combination);
        }
    }

    // 主函数，用于计算所有可能的组合
    public static List<List<Integer>> combine(int[] nums, int k) {
        List<List<Integer>> result = new ArrayList<>();
        Arrays.sort(nums); // 先对数组进行排序，方便处理重复元素
        backtrack(result, new ArrayList<>(), nums, k, 0);
        return result;
    }

    // 回溯函数，用于递归生成所有可能的组合
    private static void backtrack(List<List<Integer>> result, List<Integer> tempList, int[] nums, int remain, int start) {
        // 如果剩余需要选择的元素数量为0，说明当前组合已经满足条件，将其加入结果集
        if (remain == 0) {
            result.add(new ArrayList<>(tempList));
            return;
        }

        // 遍历数组中的元素，尝试将其加入当前组合
        for (int i = start; i < nums.length; i++) {
            // 如果当前元素与前一个元素相同，并且不是第一个元素，则跳过，避免重复组合
            if (i > start && nums[i] == nums[i - 1]) {
                continue;
            }

            // 将当前元素加入临时组合
            tempList.add(nums[i]);
            // 递归调用，继续选择下一个元素，注意remain减1，start为i+1
            backtrack(result, tempList, nums, remain - 1, i + 1);
            // 回溯，移除最后一个元素，尝试其他可能的组合
            tempList.remove(tempList.size() - 1);
        }
    }
}