递归回溯生成全部的字符串,刚开始想全找出来,然后用哈希表去找第几个数。最开始用的是set,打算set.find(s)-set.begin(),但是set容器不能相减(只有地址按顺序存放的才可以,比如vector,deque),就加了个n从开始数,数到那个就输出。但过了3/5个用例,又想到了是不是本来就是按顺序输出的,看了一下确实,就用map去存取map<string,int>直接对应第几个,但还是超时。我就发现了,不能全部生成,应该生成到输入的那个就可以停止了,加了这个限制之后果然过了。如果还没过的话,我的寻找已经是最优了,我就只能去想怎么优化生成过程了。
#include<iostream>
#include<vector>
#include<algorithm>
#include<map>
#include<math.h>
#include<set>
#include<string>
#include<stack>
using namespace std;
#include<queue>
int n = 0;//n用来记录是第几个
string ans;//需要找到的答案,也就是输入的字符串
int flag = 0;//flag标记有没有找到答案
//s是回溯生成的字符串、v是字符集‘a',’b'、used是字符集中的字符有没有被使用
void backtrack(string s, vector<char>& v, vector<int>& used)
{
if (flag == 1)
{
//如果已经找到答案了就直接返回
return;
}
if (s.size() == v.size())
{
//找到了直接输出并返回,并flag=1标记为找到了答案
if (ans == s)
{
cout << n;
flag = 1;
return;
}
n++;
return;
}
//回溯过程
for (int i = 0; i < v.size(); i++)
{
if (used[i] == 0)
{
used[i] = 1;
s += v[i];
backtrack(s,v, used);
s.erase(s.end() - 1);//删除s末尾元素
used[i] = 0;
}
}
}
int main()
{
string s;
cin >> s;
ans = s;
vector<char>tmp;
//tmp存放字符集
for (int i = 0; i < s.size(); i++)
{
char ch = 'a' + i;
tmp.push_back(ch);
}
vector<int>used(s.size(), 0);
string ss;
backtrack(ss,tmp, used);
return 0;
}