目录
一、题目描述
有重复字符串的排列组合。编写一种方法,计算某字符串的所有排列组合。
示例1:
输入:S = "qqe"
输出:["eqq","qeq","qqe"]
示例2:
输入:S = "ab"
输出:["ab", "ba"]
提示:
- 字符都是英文字母。
- 字符串长度在[1, 9]之间。
二、解题思路
暴力回溯依然可以解,时间代价较高。依然是看见了好的去重方式,所以记录下来。
使用map统计字符的出现次数,然后根据次数来回溯。
这样操作可以去重的原因在于在map里统计的只是字符的出现次数,并没有位置顺序的概念,所以自然无论有多少个重负的字符,只计算了一种结果;
例如:两个a,一个b,一个c;那么map里的数据就是{{a,2},{b,1},{c,1}};在根据出现次数回溯过程中,aabc当中的前两个a跟位置没有关系,只与次数有关;如果与位置有关就会有和的区别。
三、代码实现
#include <bits/stdc++.h>
using namespace std;
//法一;暴力回溯
unordered_set<string> tmp;
void permutation(string str, vector<bool>& used, string S) {
if (str.size() == S.size()) {
tmp.insert(str);
return;
}
for (int i = 0; i < S.size(); i++) {
if (used[i] == false) {
used[i] = true;
str += S[i];
permutation(str, used, S);
used[i] = false;
str.pop_back();
}
}
}
vector<string> permutation(string S) {
int length = S.size();
vector<bool> used(length, false);
string str = "";
permutation(str, used, S);
return vector<string>(tmp.begin(), tmp.end());
}
//法二:这种方式不好理解(来自力扣题解区)
vector<string>res;
void process(string S, string& s, int n, vector<int>&flag) {
//终止条件
if (s.size() == S.size()) {
res.push_back(s);
} else {
for (int i = 0; i < n; i++) {
if (flag[i] == 0) {
//剪枝操作
if (i > 0 && S[i] == S[i - 1] && flag[i - 1] == 1)continue;//若有重复字符,跳过该组合
s += S[i];
flag[i] = 1;
process(S, s, n, flag);
s.pop_back();//回溯到上一步
flag[i] = 0;
}
}
}
}
vector<string> permutation(string S) {
string s;
int n = S.size();
sort(S.begin(), S.end());//排序使得重复字符相邻
vector<int>flag(n);
process(S, s, n, flag);
return res;
}
//法三:使用map统计计数(来自力扣题解区)
// 执行用时:4 ms, 在所有 C++ 提交中击败了94.51%的用户
// 内存消耗:7.2 MB, 在所有 C++ 提交中击败了100.00%的用户
// 使用map去重
vector<string> permutation(string S) {
unordered_map<char, int> m;
for (auto c : S) {
if (m.count(c)) {
m[c]++;
} else {
m[c] = 1;
}
}
vector<string> permutations;
backtrack(permutations, m, "", S.size());
return permutations;
}
void backtrack(vector<string> &permutations, unordered_map<char, int> &m, string prefix, const int len) {
if (prefix.size() == len) {
permutations.push_back(prefix);
return;
}
//根据次数来回溯
for (unordered_map<char, int>::iterator it = m.begin(); it != m.end(); it++) {
if (it->second) {
//cout << it->first << " " << it->second << " " << prefix << endl;
(it->second)--; // 选择
backtrack(permutations, m, prefix + it->first, len); // 递归判定
(it->second)++; // 回溯
}
}
}
int main() {
string S = "qqe";
vector<string> res = permutation(S);
for (auto x : res) {
cout << x << " ";
}
return 0;
}