超时 未剪枝
import java.util.ArrayList;
import java.util.List;
/**
* @Description: TODO
* @author: lzm
* @date: 2021年06月22日 18:06
*/
public class Test6 {
public static void main(String[] args) {
String s = "aab";
Test6 test6 = new Test6();
String[] permutation = test6.permutation(s);
for (int i = 0; i < permutation.length; i++) {
System.out.println(permutation[i]);
}
}
public String[] permutation(String s) {
int length=s.length();
List<String> list=new ArrayList<>();
boolean[] isUse=new boolean[length];
int lever=0;
StringBuilder temp = new StringBuilder();
dfs(list,isUse,lever,length,s,temp);
String[] res=new String[list.size()];
for (int i = 0; i < list.size(); i++) {
res[i]=list.get(i);
}
return res;
}
private void dfs(List<String> list, boolean[] isUse, int lever, int length, String s, StringBuilder temp) {
if (lever==length ){
if (list.contains(temp.toString())){
return;
}
list.add(new String(temp));
return;
}
for (int i = 0; i < s.length(); i++) {
if (isUse[i]){
continue;
}
temp.append(s.charAt(i));
isUse[i]=true;
dfs(list,isUse,lever+1,length,s,temp);
isUse[i]=false;
temp.deleteCharAt(temp.length()-1);
}
}
}
剪枝
class Solution {
public String[] permutation(String s) {
int len = s.length();
if (len == 0) {
return new String[0];
}
// 转换成字符数组是常见的做法
char[] charArr = s.toCharArray();
// 排序是为了去重方便
Arrays.sort(charArr);
// 由于操作的都是字符,使用 StringBuilder
StringBuilder path = new StringBuilder();
boolean[] used = new boolean[len];
// 为了方便收集结果,使用动态数组
List<String> res = new ArrayList<>();
dfs(charArr, len, 0, used, path, res);
// 记得转成字符串数组
return res.toArray(new String[0]);
}
/**
* @param charArr 字符数组
* @param len 字符数组的长度
* @param depth 当前递归深度
* @param used 当前字符是否使用
* @param path 从根结点到叶子结点的路径
* @param res 保存结果集的变量
*/
private void dfs(char[] charArr,
int len,
int depth,
boolean[] used,
StringBuilder path,
List<String> res) {
if (depth == len) {
// path.toString() 恰好生成了新的字符对象
res.add(path.toString());
return;
}
for (int i = 0; i < len; i++) {
if (!used[i]) {
if (i > 0 && charArr[i] == charArr[i - 1] && !used[i - 1]) {
continue;
}
used[i] = true;
path.append(charArr[i]);
dfs(charArr, len, depth + 1, used, path, res);
// 递归完成以后,需要撤销选择,递归方法执行之前做了什么,递归方法执行以后就需要做相应的逆向操作
path.deleteCharAt(path.length() - 1);
used[i] = false;
}
}
}
}