【剑指Offer系列38】字符串的排列

题目

输入一个字符串，打印出该字符串中字符的所有排列。
你可以以任意顺序返回这个字符串数组，但里面不能有重复元素。

示例:
输入：s = “abc”
输出：[“abc”,“acb”,“bac”,“bca”,“cab”,“cba”]

限制：
1 <= s 的长度 <= 8

代码

Python

# 思路：
# 采用排列组合思想，回溯法+剪枝
# 复杂度：
# O(N!)
class Solution:
    def permutation(self, s: str) -> List[str]:
        c,res=list(s),[]
        def dfs(x):
            if x==len(c)-1:
                res.append(''.join(c)) # 添加排列方案
                return 
            dic=set()
            for i in range(x,len(c)): 
                if c[i] in dic: continue # 剪枝
                dic.add(c[i])
                c[i],c[x]=c[x],c[i] # 固定c[i]在x位
                dfs(x+1) # 固定x+1位
                c[i],c[x]=c[x],c[i]
        dfs(0)
        return res

C++

class Solution {
public:
    vector<string> res;
    vector<char> c;
    
    vector<string> permutation(string s) {
        for (int i=0;i<s.size();i++) {
            c.push_back(s[i]);
        }
        dfs(0);
        return res;
    }
    
    void dfs(int x) {
        if (x==c.size()-1) {
            string ans(c.begin(),c.end());
            res.push_back(ans);
            return ;
        }
        set<char> dic;
        for (int i=x;i<c.size();i++) {
            if (dic.count(c[i])) continue;
            dic.insert(c[i]);
            swap(c[i],c[x]);
            dfs(x+1);
            swap(c[i],c[x]);
        }
    }
};