leetcode784 字母大小写全排列

Posted 2020-11-25 patrolli

题目：https://leetcode-cn.com/problems/letter-case-permutation/

给定一个字符串S，通过将字符串S中的每个字母转变大小写，我们可以获得一个新的字符串。返回所有可能得到的字符串集合。

样例输入与输出：

S = "a1b2"

["a1b2", "a1B2", "A1b2", "A1B2"]

S = "3z4"

["3z4", "3Z4"]

思路：

• 画出解答树，对树做深度优先搜索。ASCII码中，大写字母和小写字母差为32（小写比大写大32），这里可以直接将字母与32做异或，达到大写小写的转换操作。

解答树为：

代码：

class Solution {
public:
    vector<string> letterCasePermutation(string S) {
        vector<string> ans;
        string tmp;
        dfs(S, 0, ans);
        return ans;
    }
private:
    void dfs(string S, int idx, vector<string>& ans){
        if(idx == S.size()){
            ans.push_back(S);
            return;
        }
        dfs(S, idx+1, ans);
        if(!isalpha(S[idx])) return;
        S[idx] ^= (1<<5);
        dfs(S, idx+1, ans);
        S[idx] ^= (1<<5);
    }
};