剑指 38. 字符串的排列

题目描述

输入一个字符串，打印出该字符串中字符的所有排列。

你可以以任意顺序返回这个字符串数组，但里面不能有重复元素。

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

class Solution {
public:
    vector<string> ans;
    vector<bool> flag;
    vector<string> permutation(string s) {
        sort(s.begin(),s.end());
        flag=vector<bool>(s.size(),true);
        dfs(s,"");
        return ans;
    }
    
    void dfs(string& s, string res){
        if(res.size()==s.size()){
            ans.push_back(res);
            return;
        }
        
        for(int i=0;i<s.size();i++){
            if(flag[i]){
                flag[i]=false;
                dfs(s,res+s[i]);
                flag[i]=true;
                while(i+1<s.size()&&s[i+1]==s[i]) i++;//关键点
            }
        }
        return;
    }
};

总结

该题的一个难点是当序列有相同元素时如何去除重复序列？如aaab。

首先对字符串进行排序，使得相同字母相邻，在dfs返回后，我们需要继续遍历i+1的位置时，需要判断i+1是否与i相同，如果相同则跳过。

假如目前s=“aaab”，长度为4，返回，s=“aaa”，i=3，此时for循环结束，返回s=“aa”，i=2，继续循环，s=“aaba”，返回，s=“aab”，i=3，返回，s=“aa”，i=3，返回，s=“a”，i=1，当遍历i=2时，得到的字符串s=“aa”，与之前遍历i=1时得到的字符串重复，因此舍弃i=2,遍历i=3，即s=“ab”.