算法学习--全排列

//全排列
#include <iostream>
#include <algorithm>
#include <sstream>
#include <cstring>
#include <string>
#include <vector>
using namespace std;

//统计元素在string中出现的次数
int cal_num(char ch, string str) {
	int cnt = 0;
	for (int i = 0; i < str.length(); i++) {
		if (ch == str[i])
			cnt++;
	}
	return cnt;
}

//递归实现
//len为长度
vector<string> getAllpai_1(string str, int len) {
	vector<string> res;//建立一个空集保存结果
	if (len == 1) {
		stringstream ss;
		ss << str[0];
		res.push_back(ss.str());
		return res;
	}
	vector<string> result_last = getAllpai_1(str, len - 1);//上一层的结果集
	for (string s : result_last) {
		//1.新元素插在最前面
		res.push_back(str[len - 1] + s);
		//2.新元素插在最后面
		res.push_back(s + str[len - 1]);
		//3.新元素插在中间（j表示新元素插入的位置）
		for (int j = 1; j < s.length(); j++) {
			res.push_back(s.substr(0, j) + str[len - 1] + s.substr(j));
		}
	}
	return res;
}

//迭代实现
vector<string> getAllpai_2(string str) {
	vector<string> result_old;
	//初始化结果集，包含第一个字符（这里注意单个字符转字符串的方法）
	stringstream ss;
	ss << str.at(0);
	result_old.push_back(ss.str());
	//从第二个元素开始，遍历每一个元素
	for (int i = 1; i < str.length(); i++) {
		vector<string> result_new;//注意每次的新集合都是空的（不然有很多子集出来了）
		for (string s : result_old) {
			//1.新元素插在最前面
			result_new.push_back(str.at(i) + s);
			//2.新元素插在最后面
			result_new.push_back(s + str.at(i));
			//3.新元素插在中间（j表示新元素插入的位置）
			for (int j = 1; j < s.length(); j++) {
				result_new.push_back(s.substr(0, j) + str.at(i) + s.substr(j));
			}
		}
		result_old = result_new;
	}
	return result_old;
}

//交换，回溯（简洁）
//v_res为保存结果的容器
void getAllpai_3(vector<string> &v_res, string str, int index) {
	if (index == str.length() - 1) {
		v_res.push_back(str);
		return;
	}
	for (int i = index; i < str.length(); i++) {
		swap(str[index], str[i]);
		getAllpai_3(v_res, str, index + 1);
		swap(str[index], str[i]);
	}
}

//前缀法（用于求全排列中按字典序排序的第k个排列，数据量大的时候显得效率较高）
//prefix为当前的前缀,用count来计数
int count_2 = 0;
void getAllpai_4(string str, string prefix, int k) {
	if (prefix.length() == str.length()) {
		count_2++;
		if (count_2 == k) {
			cout << prefix << endl;
			exit(0);
		}
	}
	//每次都从头开始扫描，只要该字符可用，就附加到前缀后面
	for (int i = 0; i < str.length(); i++) {
		char ch = str[i];
		//这个字符可用的条件：在prefix中出现的次数小于在字符集中出现的次数
		//这个其实是避免有元素重复
		if (cal_num(ch, prefix) < cal_num(ch, str)) {
			getAllpai_4(str, prefix + ch, k);
		}
	}
}


int main() {
	string str = "adbec";
	int count = 1;
	//vector<string> res = getAllpai_1(str, str.length());
	//for (string s : res) {
	//	cout << count++ << ":" << s << endl;
	//}
	count = 1;
	cout << "交换回溯法结果：--------------------------------" << endl;
	vector<string> v_res;
	getAllpai_3(v_res, str, 0);
	sort(v_res.begin(), v_res.end());
	for (string s : v_res) {
		cout << count++ << ":" << s << endl;
	}
	int k;
	cin >> k;
	cout << "全排列中按字典序排序的第" << k << "个排列:";
	sort(str.begin(), str.end());
	getAllpai_4(str, "", k);
	return 0;
}