题目要求:给定一个字符串,通过将其中的每个字符转换大小写得到一个新的字符串,要求返回所有可能的字符串集合。
思路:我们可以利用回溯加上深度优先搜索(dfs)的思路来解答这一题。我们先用一个for循环对字符串的所有字母元素进行遍历,以遍历到的每一个字符做为基点,接着对当前位置的后续字符进行大小写转换并记住其排列组合,直到达到最后一个字符。
比如字符串3ae4第一次进入递归时会将a变为A,而其余的字符不变,这样就记录了一种可能。我们可以观察发现循环中的每一次递归都会为原本的字符串设置一种初始状态,而后续的递归则会每次以当前状态为基准对后续的字符进行排列。如图:
class Solution {
// 由于不记得大小字母转换的ascii插值是多少,所以将其定为一个常量
final int DIFF = 'a' - 'A';
public List<String> letterCasePermutation(String s) {
List<String> result = new ArrayList<>(); // 结果集
result.add(s); // 未遍历之前也是一种可能
char[] chars = s.toCharArray();
for (int i = 0; i < chars.length; i++) {
// 遍历每个字符,从该字符开始依次枚举其排列组合
char c = chars[i];
if (c >= 'A' && c <= 'Z' ||
c >= 'a' && c <= 'z') {
dfs(result, i, chars);
}
}
return result;
}
private void dfs(List<String> result, int idx, char[] chars) {
if (idx == chars.length) {
// 递归结束条件,即索引和字符数组长度相等
return;
}
// 转变对应位置字母的大小写
char c = chars[idx];
if (c >= 'A' && c <= 'Z') {
chars[idx] += DIFF;
} else {
chars[idx] -= DIFF;
}
// 记录结果
result.add(new String(chars));
// 找到下一个字母的位置
for (int i = idx + 1; i < chars.length; i++) {
char next = chars[i];
if (next >= 'A' && next <= 'Z' ||
next >= 'a' && next <= 'z') {
// 递归调用函数对下一个字符做处理
dfs(result, i, chars);
}
}
// 回溯
if (chars[idx] >= 'A' && chars[idx] <= 'Z') {
chars[idx] += DIFF;
} else {
chars[idx] -= DIFF;
}
}
}