题解 | #递归实现组合型枚举#
递归实现组合型枚举
https://ac.nowcoder.com/acm/problem/50918
#include <stdio.h> #include <stdlib.h> int compare(const void *a, const void *b) { return (*(int*)a - *(int*)b); } void printCombination(int arr[], int n, int m, int index, int data[], int i) { if (index == m) { for (int j = 0; j < m; j++) { printf("%d ", data[j]); } printf("\n"); return; } if (i >= n) { return; } data[index] = arr[i]; printCombination(arr, n, m, index + 1, data, i + 1); while (i < n - 1 && arr[i] == arr[i + 1]) { i++; } printCombination(arr, n, m, index, data, i + 1); } void printAllCombinations(int n, int m) { int arr[n]; for (int i = 0; i < n; i++) { arr[i] = i + 1; } int data[m]; qsort(arr, n, sizeof(int), compare); printCombination(arr, n, m, 0, data, 0); } int main() { int n, m; scanf("%d %d", &n, &m); printAllCombinations(n, m); return 0; }
这段代码使用了递归函数printCombination来生成所有可能的选择方案。它首先对输入数组进行排序,然后调用递归函数printCombination来生成所有可能的组合。递归函数根据索引和当前选择的元素个数,分别递增索引和选择下一个元素。当选择的元素个数达到目标值时,它将打印出这个组合。然后,它将跳过重复的元素,并继续生成其他组合。
在main函数中,我们首先读取输入的n和m,然后调用printAllCombinations函数来生成并打印所有可能的选择方案。