其他分享
首页 > 其他分享> > AcWing 190.字串变换

AcWing 190.字串变换

作者:互联网

很明显是最小步数模型,我们先来分析一波单向起点开始\(bfs\)需要的空间:

假设每次决策数量是 \(K\),那么如果直接\(bfs\),最坏情况下的搜索空间是 \(K^10\),非常大,所以会\(TLE\)或者\(MLE\)。第一层是\(1\),第二层是\(K\),第三层是\(K*K=K^2…..K^10\)

如果采用双向\(bfs\),则可以把搜索空间降到 \(2 \times K^5\)。在实际测试中只需 \(20ms\) 左右,剪枝效果很好。

\(bfs\)的扩展方式是:

双向\(bfs\)的一个优化:在双向\(bfs\)时,每次选择队列中元素数量较少的方向来扩展。

总结:

#include <bits/stdc++.h>

using namespace std;
const int INF = 0x3f3f3f3f;
//双向bfs a<->b
//每次需要扩展出一层,不能只扩展出一个,时间复杂度可以优化  k^10-> k^5 *2
//大概估计一下,如果单向容易超时,再改双向尝试。
const int N = 6;
int n;
string a[N], b[N];
int Min = INF; //最短步数
// 功能:取出队头元素,进行n种规则扩展
// q :需要扩展的队列
// da:到起点的距离
// db:到终点的距离
// a :转换前的字符串
// b :转换后的字符串
//返回值:true false,本轮扩展,是否找到了最短步数
bool extend(queue<string> &q, unordered_map<string, int> &da, unordered_map<string, int> &db,
            string a[N], string b[N]) {
    // 取出队头元素
    string t = q.front();
    q.pop();

    for (int i = 0; i < t.size(); i++) // 出发串的每个字符
        for (int j = 0; j < n; j++)    // 枚举规则
            //如果t这个字符串的一段= 规则,比如= xyz,才可以替换
            if (t.substr(i, a[j].size()) == a[j]) {
                // 变换之后的结果r:前面不变的部分+ 变化的部分 + 后面不变的部分
                // 比如abcd ,根据规则abc--> xu,变成 xud,这里的r就是xud
                string r = t.substr(0, i) + b[j] + t.substr(i + a[j].size());
                // r状态是否落到b里面去,两个方向会师,返回最小步数
                if (db.count(r)) {
                    Min = da[t] + 1 + db[r];
                    return true;
                }
                // 如果该状态之前已扩展过,
                if (da.count(r)) continue;
                //没有扩展过,需要加入队列
                da[r] = da[t] + 1;
                q.push(r);
            }
    return false;
}
// 从起点和终点来做bfs
void bfs(string A, string B) {
    if (A == B) {
        Min = 0;
        return;
    } //加入特判,一样了还变换个啥

    queue<string> qa, qb;              //两个方向的队列
    unordered_map<string, int> da, db; //每个状态到起点的距离da(哈希表),每个状态到终点的距离db哈希表

    // qa从起点开始搜,qb从终点开始搜
    qa.push(A), da[A] = 0; // 起点A到起点的距离为0
    qb.push(B), db[B] = 0; // 终点B到终点B的距离为0

    while (qa.size() && qb.size()) {
        bool finish = false;
        // 哪个方向的队列的长度更小一些,空间更小一些,从该方向开始扩展,时间复杂度比较平滑,否则有1个点会超时
        if (qa.size() <= qb.size())
            finish = extend(qa, da, db, a, b);
        else
            finish = extend(qb, db, da, b, a); //以b为起点理解,那么da(终点),db(起点)的概念是需要转化的,同理规则也需要b,a转化

        //扩展后找到最小步数就结束
        if (finish) return;
    }
}

int main() {
    string A, B;
    cin >> A >> B;
    // 读入扩展规则,分别存在a数组和b数组
    // 本题没有给出规则的个数,需要我们用while(cin>>str)读取,vscode中以ctrl+z结束输入。
    while (cin >> a[n] >> b[n]) n++; // n记录的是规则数
    bfs(A, B);

    if (Min > 10)
        puts("NO ANSWER!");
    else
        cout << Min << endl;
}

标签:string,db,扩展,190,da,bfs,字串,size,AcWing
来源: https://www.cnblogs.com/littlehb/p/15963185.html