【深度遍历】【排列组合】【力扣】有重复字符串的排列组合

题目描述

有重复字符串的排列组合。编写一种方法，计算某字符串的所有排列组合。

示例1:

复制代码

 输入：S = "qqe"
 输出：["eqq","qeq","qqe"]

示例2:

复制代码

 输入：S = "ab"
 输出：["ab", "ba"]

提示:

字符都是英文字母。
字符串长度在 $1, 9$ 之间。

解题思路

排列组合用深度遍历即可获取全部情况

注意点：重复的组合出现一次即可，比如 $e, 第1个q, 第2个q$ 和 $e, 第2个q, 第1个q$ 为同一个排列组合

解答（C语言）

cpp 复制代码

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

char **result = NULL;  // 保存全部情况
int count = 0;  // 情况个数

void Find(int pos, char *S, int isUsed[], char dest[], size_t sLen)
{
    // pos的含义：尝试从全部字母中取出一个字母放到排列组合的第pos个位置

    // 刚好组合完成
    if (pos == sLen) {
        // printf("%s\n", dest);

        // 保存结果
        count++;
        result = (char **)realloc(result, count * sizeof(char *));
        result[count - 1] = (char *)malloc(sLen + 1);
        strcpy(result[count - 1], dest);

        return;
    }

    // 从剩余的字母中取出一个字母
    for (int i = 0; i < sLen; i++) {
        // 已经被用来组合的字母不再使用
        if (isUsed[i] == 1) {
            continue;
        }

        // 两个相同字母出现一次即可
        if (i+1 < sLen && S[i] == S[i+1] && isUsed[i+1] == 1) {
            continue;
        }

        // 因为第i个字母未被使用，因此取出来放到排列组合的第pos个位置
        dest[pos] = S[i];
        // 第i个字母标记为已使用
        isUsed[i] = 1;

        // 对排列组合的第pos个位置的下一个位置找字母
        Find(pos + 1, S, isUsed, dest, sLen);

        // 排列组合的第pos个位置尝试过第i个字母
        // 现在不用第i个字母了，而是继续循环尝试第i+1个字母
        isUsed[i] = 0;
    }

    /* 
    循环结束时函数也会结束，
    说明在第pos这个深度处的全部遍历结束，
    在 pos - 1深度处的遍历中，对下一个字母的尝试将开始
    */
}

int CmpFunc(const void *a, const void *b)
{
    return *(char *)a - *(char *)b;
}

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
char **permutation(char *S, int *returnSize)
{
    count = 0;
    result = NULL;

    // 准备要用到的变量
    size_t sLen = strlen(S);

    char dest[sLen + 1];
    memset(dest, 0, sizeof(dest));

    int isUsed[sLen];
    memset(isUsed, 0, sizeof(isUsed));

    // 排序
    qsort(S, sLen, sizeof(char), CmpFunc);

    // 深度遍历
    Find(0, S, isUsed, dest, sLen);

    // 返回结果
    *returnSize = count;

    return result;
}

int main() 
{
    int select = 1;
    if (select == 1) {
        char S[] = "qqe";
        int returnSize;
        permutation(S, &returnSize);
    } else if (select == 2) {
        char S[] = "ab";
        int returnSize;
        permutation(S, &returnSize);
    } 
   
    return 0;
}

解答（python）

python 复制代码

class Solution:
    def permutation(self, S: str) -> List[str]:
        data = list(S)
        data.sort()
        used = [False] * len(data)
        res = []

        def backtrack(content):
            if len(content) == len(data):
                res.append(content)
                return
            
            for i in range(len(data)):
                # 当前轮组合中已使用过不再使用
                if used[i]:
                    continue
                
                # 使用过的相邻元素不再使用
                if i > 0 and data[i] == data[i-1] and not used[i-1]:
                    continue
                
                # 将当前组合使用起来
                content += data[i]
                used[i] = True

                # 继续下一轮的取数
                backtrack(content)

                # 当组合取数完毕或者循环完毕，释放当前位置数的可使用状态
                used[i] = False
                content = content[:-1]   # 重新从该位置取数组合时，应当将content恢复到该位置时的状态               


        backtrack('')

        return res