字串变换
题目描述
核心思路
用双向BFS。思路如下:
- 从起点和终点同时进行BFS。
- 同时定义两个队列, 哪个队列小,就对哪个队列进行扩展(bfs)
- 扩展过程中, 要判断另外个队列是否已经包含了当前新扩展的Key
- 如果包含, 直接return 两个距离之和
- 如果不包含, 继续扩展
扩展的大致过程如下图:
代码
#include<iostream>
#include<cstring>
#include<unordered_map>
#include<queue>
using namespace std;
const int N = 6;
int n;//n是转换规则的个数
string a[N], b[N];
//根据转换规则,将a串转成b串
int extend(queue<string>& q, unordered_map<string, int>& da, unordered_map<string, int>& db, string a[], string b[])
{
string t = q.front();//取出队头元素
q.pop();//让队头元素出队
//枚举队头元素的这个字符串的每个字符,找到t串中的哪些子串可以转换
for (int i = 0; i < t.size(); i++)
{
for (int j = 0; j < n; j++)//枚举n种转换规则
{
//t.substr(i, a[j].size())表示t中的a[i]~a[a[j].size()]这个子串可以转换 a[j]是输入规定的可以转换的子串
if (t.substr(i, a[j].size()) == a[j])
{
//t.substr(0, i)是前部分 b[j]是转换后的子串
//t.substr(i + a[j].size())从数组下标为i + a[j].size()开始的后部分
string state = t.substr(0, i) + b[j] + t.substr(i+ a[j].size());
//如果当前state这个字符串的状态已经被遍历过了,则跳过,继续
if (da.count(state))
continue;
//如果当前从起点开始搜索的state这个字符串的状态已经被"从终点开始搜索的"遍历过了
//这说明state这个字符串就是"从起点开始搜索"和"从终点开始搜索"的相遇点
//类似于打隧道,B已经打通了state,此时只要A打破state,那么就把隧道都打通了
if (db.count(state))
return da[t] + db[state] + 1;//因为A打破state需要1步,所以+1
//到这里说明state这个字符串的状态还没有被遍历过,它的步数是在t的步数的基础上+1
da[state]=da[t] + 1;
q.push(state);//将state这个字符串的状态加入队列
}
}
}
return 11;//说明在10步以内不能转换,那就随便返回一个大于10的数
}
int bfs(string A, string B)
{
queue<string>qa, qb;//队列qa来存储从起点拓展的状态,队列qb存储从终点拓展的状态
//da表示从起点拓展到某一层用了多少步数 db表示从终点拓展到某一层用了多少步数
unordered_map<string, int>da, db;
qa.push(A);//将起点的字符串的这个状态加入队列
qb.push(B);//将终点的字符串的这个状态加入队列
da[A] = 0;//起点还不需要拓展,所以步数为0
db[B] = 0;//终点还不需要拓展,所以步数为0
//如果出现队列为空,说明这这两个队列没有交集,这个图不是连通的
//就跟打隧道一样,隧道为100米,A打通了40米就为停了,B打通了50米就为停了,那么还有10米,所以不可能打通
while (qa.size() && qb.size())
{
int t;//最少的变换步数
//这里有个优化就是要优先考虑队列中状态个数少的
if (qa.size() <= qb.size())
t = extend(qa, da, db, a, b);
else
t = extend(qb, db, da, b, a);//这里传参要注意是从终点开始传递
if (t <= 10)
return t;
}
return 11;//说明在10步以内不能转换,那就随便返回一个大于10的数
}
int main()
{
string A, B;//A是起点字符串,B是终点字符串
cin >> A >> B;
//因为这里并没有明确告诉转换规则由多少个,所以需要根据输入的转换字符串来确定转换规则的个数
while (cin >> a[n] >> b[n])
n++;
int step = bfs(A, B);
if (step > 10)
puts("NO ANSWER!");
else
printf("%d\n", step);
return 0;
}