我正在尝试编写一个程序,在存储在几个数组中的单词列表上生成排列。
例如,我的程序要求两组单词,如下所示:
words #1: abc def ghi
words #2: 123 456
我想要的是这个输出:
abc 123 | abc 456 | def 123 | def 456 | ghi 123 | ghi 456
或:
123 abc | 123 def | 123 ghi | 456 abc | 456 def | 456 ghi
顺序无关紧要。
我可能会制作一个大小不固定的单词数组。那么输入将是:
words #1: abc def ghi
words #2: 123 456
words #3: ** --
以及输出:
abc 123 ** | abc 123 -- | abc 456 ** | abc 456 -- | def 123 ** | def 123 -- | def 456 ** | def 456 -- | ghi 123 ** | ghi 123 -- | ghi 456 ** | ghi 456 --
我想我不得不考虑使用递归函数,但我有点困惑。
这是我写的:
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
typedef struct permut_word_s {
char *str;
} permut_word_t;
typedef struct permut_group_s {
permut_word_t *words;
int nb_words;
} permut_group_t;
static int split(char *str,
char *token,
permut_group_t *g) {
permut_word_t *a = NULL;
permut_word_t *t = NULL;
char *p = NULL;
int nbw = 0;
int l = 0;
if(!str || !token || !g) {
return -1;
}
p = strtok(str, token);
while(p != NULL) {
if(!(t = realloc(a, (nbw + 1) * sizeof(permut_word_t)))) {
return -1;
}
if(!(t[nbw].str = malloc(strlen(p) + 1))) {
return -1;
}
memset(t[nbw].str, 0, strlen(p) + 1);
if(!(strncpy(t[nbw].str, p, strlen(p)))) {
return -1;
}
nbw++;
p = strtok(NULL, token);
a = t;
}
g->words = a;
g->nb_words = nbw;
return 0;
}
void word_free(permut_word_t *w) {
if(!w) {
return;
}
if(w->str) {
free(w->str);
}
return;
}
void group_free(permut_group_t *g) {
int i = 0;
if(!g) {
return;
}
for(; i < g->nb_words; i++) {
if(&g->words[i]) {
word_free(&g->words[i]);
}
}
free(g->words);
return;
}
void permut(permut_group_t *g,
int cur,
int len) {
int i = 0;
int j = 0;
if(cur == len) {
return;
}
for(i = cur; i < len; i++) {
for(j = 0; j < g[cur].nb_words; j++) {
printf("%s ", g[cur].words[j].str);
}
permut(g, cur + 1, len);
}
}
int main(int argc, char **argv) {
char buf[1024] = { 0 };
permut_group_t *groups = NULL;
int len = 0;
(void)argc;
(void)argv;
if(!(groups = malloc(2 * sizeof(permut_group_t)))) {
return -1;
}
fprintf(stdout, "words #1: ");
fgets(buf, 1024, stdin);
split(buf, " ", &groups[0]);
len++;
fprintf(stdout, "words #2: ");
fgets(buf, 1024, stdin);
split(buf, " ", &groups[1]);
len++;
permut(&groups[0], 0, len);
group_free(&groups[0]);
group_free(&groups[1]);
free(groups);
return 0;
}
如果知道groups数组的大小可能是可变的,它怎么能正确地做到呢?
最佳答案
请尝试以下操作:
void permut(permut_group_t *g,
int cur,
int len, char *result=NULL) {
int j = 0;
if(cur == len) {
printf("%s\n", result);
return;
}
for(j = 0; j < g[cur].nb_words; j++) {
char nextLine[200];
if (cur==0)
strncpy (nextLine, g[cur].words[j].str, 200);
else
snprintf (nextLine, 200, "%s;%s", result, g[cur].words[j].str);
permut(g, cur + 1, len, nextLine);
}
}
以下输入
words #1: AB CD
words #2: 11 22 33
然后生成以下输出:
AB;11
AB;22
AB;33
CD;11
CD;22
CD;33
诀窍在于,在收集结果时,必须保留中间获得的组合,并将其传递到下一个级别。因此,函数
permut
的签名被一个(可选)参数char* result
扩展,该参数充当这些中间(或最终)组合的“存储器”。置换函数的签名现在是
permut(permut_group_t *g, int cur, int len, char *result=NULL)
;从函数main
开始,可以保持调用permut(&groups[0], 0, len)
不变,因为当参数result
为0时,可以省略可选参数cur
;请注意,我还扩展了函数
split
,以便删除任何后续的新行,从而不会将这些新行复制到结果中:char *newLine = strrchr(str, '\n');
if (newLine)
*newLine = '\0';