JZ27 剑指offer 字符串的排列

最新推荐文章于 2021-07-22 14:27:59 发布

原创最新推荐文章于 2021-07-22 14:27:59 发布 · 107 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

剑指

本文介绍了如何使用Python实现字符串的所有排列，并探讨了利用字典和剪枝优化递归的方法。通过实例演示了Solution类中的permutation函数，以及两种版本的代码对比，突出了剪枝策略在避免重复的重要性和效率提升。

第27题字符串的排列

题目描述

输入一个字符串,按字典序打印出该字符串中字符的所有排列。例如输入字符串abc,则按字典序打印出由字符a,b,c所能排列出来的所有字符串abc,acb,bac,bca,cab和cba。

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

python版本，每个内层加个只属于当前层的字典，遍历到重复字母进行continue操作，进行剪枝

class Solution {
public:
    set<string> ret;
    void perm(int pos,string s){
        if(pos+1 == s.length()){
            ret.insert(s);
            return;
        }
        for(int i = pos;i<s.length();++i){
            swap(s[pos],s[i]);
            perm(pos+1,s);
            swap(s[pos],s[i]);
        }
    }
    vector<string> Permutation(string str) {
        if(str.empty()){
            return {};
        }
        perm(0,str);
        return vector<string>({ret.begin(),ret.end()});
    }
};

思路：
对于递归函数perm，传入当前位置pos和字符串s，从pos开始遍历，每次将pos位和当前遍历i位的字符交换（相当于固定当前遍历位置，因为交换后我们只考虑pos+1后的进行递归），然后从pos+1开始递归，直到当前位置为字符串末位停止（实际上在遍历到字符串倒数第二位时，循环进行两次，一次不动，一次与末位交换，每次交换后进入递归后就会插入set并返回，所以在末位不进行循环），放入哈希表中防止重复。每次递归回来时要在进行一次swap操作交换回当前遍历之前状态。