C++ 搜索之广度优先搜索(BFS)

例题1:

附C++代码:

//
//  1456.cpp
//  胜利大逃亡
//
//  Created by chenmeiqi on 2019/4/11.
//  Copyright © 2019年 chenmeiqi. All rights reserved.
//

#include <iostream>
#include <queue>
using namespace std;
struct N{               // 状态结构体
    int x,y,z;
    int t;
};
bool mark[50][50][50];      // 标记数组
int maze[50][50][50];       // 保存立方体信息
queue<N> Q;             // 队列,队列中的元素为状态
int go[][3]={           // 坐标变换数组,由坐标(x,y,z)扩展得到的新坐标均可通过(x+go[i][0],y+go[i][1],z+go[i][2])得到
    1,0,0,
    -1,0,0,
    0,1,0,
    0,-1,0,
    0,0,1,
    0,0,-1
};
int BFS(int a, int b, int c){       // 广度优先搜索,返回其最少耗时
    int res;
    while(!Q.empty()){      // 当队列中仍有元素可以扩展时循环
        N now=Q.front();        // 得到队头状态
        Q.pop();                // 从队列中弹出队头状态
        for(int i=0;i<6;i++){           // 依次扩展其六个相邻节点
            int nx=now.x+go[i][0];
            int ny=now.y+go[i][1];
            int nz=now.z+go[i][2];
            if(nx<0||nx>=a||ny<0||ny>=b||nz<0||nz>=c){      // 新坐标在立方体外,丢弃
                continue;
            }
            if(mark[nx][ny][nz]==true){         // 该位置为墙
                continue;
            }
            if(maze[nx][ny][nz]==1){        // 包含该坐标的状态已经被得到过
                continue;
            }
            N n;        // 新的状态
            n.x=nx;
            n.y=ny;
            n.z=nz;
            n.t=now.t+1;        // 新状态耗时
            mark[nx][ny][nz]=true;      // 标记该坐标
            Q.push(n);          // 将该状态放入队列
            if(nx==a-1&&ny==b-1&&nz==c-1){          // 该坐标即为终点
                return n.t;
            }
        }
    }
    return -1;
}
int main(int argc, const char * argv[]) {
    int k;
    int a,b,c,T;
    cin>>k;
    while(k--){
        cin>>a>>b>>c>>T;
        for(int  i = 0; i < a; i++)
        {
            for(int  j = 0; j < b; j++)
            {
                for(int k = 0; k < c; k++){
                    cin>>maze[i][j][k];         // 输入立方体信息
                    mark[i][j][k]=false;            // 初始化标记数组
                }
            }
            
        }
        while(!Q.empty()){      // 清空队列
            Q.pop();
        }
        mark[0][0][0]=true;     // 标记起点
        N tmp;
        tmp.x=tmp.y=tmp.z=0;
        tmp.t=0;                // 初始状态
        Q.push(tmp);
        int res=BFS(a,b,c);         // 广度优先搜索
        if(res<=T){             // 所需时间符合条件
            cout<<res<<endl;
        }
        else{
            cout<<-1<<endl;
        }
    }
    
    return 0;
}

我们通过状态之间的相互扩展完成在所有状态集合中的搜索,并查找我们需要的状态。利用这种手段,我们将原本对路径的搜索转化到了对状态的搜索上来。广度优先搜索即对由状态间的相互转移构成的解答树进行的按层次遍历。为了更好地了解状态的含义,我们来看如下有趣的问题:

例题2:

附代码:

//
//  1457.cpp
//  非常可乐
//
//  Created by chenmeiqi on 2019/4/11.
//  Copyright © 2019年 chenmeiqi. All rights reserved.
//

#include <iostream>
#include <queue>
#define S 101
using namespace std;
struct N            // 状态结构体
{
    int s,n,m;          // 每个杯子中可乐的体积
    int t;          // d得到该体积组倾倒次数
};
int mark[S][S][S];      // 对体积组(x,y,z)进行标记,即只有第一次得到包含体积组(x,y,z)的状态为有效状态,其余的舍去
queue<N> Q;     // 队列
void AtoB(int &a,int sa,int &b,int sb){     // 倾倒函数,由容积为sa的杯子倒往容积为sb的杯子,其中引用参数a和b,初始时为原始杯子中可乐的体积,当函数调用完毕后,为各自杯中可乐的新体积
    if(sb-b>=a){            // 若a可以全部倒到b中
        b+=a;
        a=0;
    }
    else{       // 否则
        a-=sb-b;
        b=sb;
    }
}
int BFS(int s, int n, int m ){
    while(!Q.empty()){          // 当队列非空时,重复循环
        N now =Q.front();       // 拿出队头状态
        Q.pop();        // 弹出队头状态
        
        int a,b,c;      // a,b,c临时保存三个杯子中可乐体积
        
        a=now.s;
        b=now.n;
        c=now.m;
        AtoB(a,s,b,n);          // 由a倒向b
        if(mark[a][b][c]==false){       // 若该体积组尚未出现
            mark[a][b][c]=true;     // 标记该体积组
            N tmp;
            tmp.s=a;
            tmp.n=b;
            tmp.m=c;
            tmp.t=now.t+1;          // 生成新的状态
            if(a==s/2 && b==s/2){           // 若该状态已经为平分状态,则直接返回该状态的耗时
                return tmp.t;
            }
            if(a==s/2 && c==s/2){
                return tmp.t;
            }
            if(b==s/2 && c==s/2){
                return tmp.t;
            }
            Q.push(tmp);            // d否则放入队列
        }
        
        a=now.s;                // 重置 a,b,c 为未倾倒前的体积
        b=now.n;
        c=now.m;
        AtoB(b,n,a,s);          // 由b倒向a
        if(mark[a][b][c]==false){
            mark[a][b][c]=true;
            N tmp;
            tmp.s=a;
            tmp.n=b;
            tmp.m=c;
            tmp.t=now.t+1;
            if(a==s/2 && b==s/2){
                return tmp.t;
            }
            if(a==s/2 && c==s/2){
                return tmp.t;
            }
            if(b==s/2 && c==s/2){
                return tmp.t;
            }
            Q.push(tmp);
        }
        
        a=now.s;
        b=now.n;
        c=now.m;
        AtoB(a,s,c,m);       // 由a倒向c
        if(mark[a][b][c]==false){
            mark[a][b][c]=true;
            N tmp;
            tmp.s=a;
            tmp.n=b;
            tmp.m=c;
            tmp.t=now.t+1;
            if(a==s/2 && b==s/2){
                return tmp.t;
            }
            if(a==s/2 && c==s/2){
                return tmp.t;
            }
            if(b==s/2 && c==s/2){
                return tmp.t;
            }
            Q.push(tmp);
        }
        
        a=now.s;
        b=now.n;
        c=now.m;
        AtoB(c,m,a,s);           // 由c倒向a
        if(mark[a][b][c]==false){
            mark[a][b][c]=true;
            N tmp;
            tmp.s=a;
            tmp.n=b;
            tmp.m=c;
            tmp.t=now.t+1;
            if(a==s/2 && b==s/2){
                return tmp.t;
            }
            if(a==s/2 && c==s/2){
                return tmp.t;
            }
            if(b==s/2 && c==s/2){
                return tmp.t;
            }
            Q.push(tmp);
        }
        
        a=now.s;
        b=now.n;
        c=now.m;
        AtoB(b,n,c,m);              // 由b倒向c
        if(mark[a][b][c]==false){
            mark[a][b][c]=true;
            N tmp;
            tmp.s=a;
            tmp.n=b;
            tmp.m=c;
            tmp.t=now.t+1;
            if(a==s/2 && b==s/2){
                return tmp.t;
            }
            if(a==s/2 && c==s/2){
                return tmp.t;
            }
            if(b==s/2 && c==s/2){
                return tmp.t;
            }
            Q.push(tmp);
        }
        
        a=now.s;
        b=now.n;
        c=now.m;
        AtoB(c,m,b,n);           // 由c倒向b
        if(mark[a][b][c]==false){
            mark[a][b][c]=true;
            N tmp;
            tmp.s=a;
            tmp.n=b;
            tmp.m=c;
            tmp.t=now.t+1;
            if(a==s/2 && b==s/2){
                return tmp.t;
            }
            if(a==s/2 && c==s/2){
                return tmp.t;
            }
            if(b==s/2 && c==s/2){
                return tmp.t;
            }
            Q.push(tmp);
        }
    }
    return -1;
}
int main(int argc, const char * argv[]) {
    int s,n,m;
    while(cin>>s>>n>>m){
        if(s==0&&n==0&&m==0){
            break;
        }
        if (s%2==1) {               // 若s为奇数则不可能平分,直接输出NO
            cout<<"NO"<<endl;
            continue;
        }
        for(int i=0;i<s;i++){               // 初始化状态
            for(int j = 0; j < n; j++)
            {
                for(int k = 0; k < m; k++)
                {
                    mark[i][j][k]=false;
                }
            }
        }
        
        while(!Q.empty()){          // 清空队列中状态
            Q.pop();
        }
        
        N begin;            // 初始时状态
        begin.s=s;
        begin.n=begin.m=begin.t=0;
        mark[s][0][0]=true;
        Q.push(begin);    // 将初始状态放入队列
        
        int res=BFS(s,n,m);         // 广度优先搜索
        if(res==-1){            // 若为-1输出NO
            cout<<"NO"<<endl;;
        }
        else{                   // 否则输出答案
            cout<<res<<endl;
        }
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值