P1811 最短路 题解（bfs）

题目描述

题目链接
给定一个包含N个点，M条边的无向图，每条边的边权均为1。

再给定K个三元组（A，B，C），表示从A点走到B点后不能往C点走。注意三元组是有序的，如可以从B点走到A点再走到C。

现在你要在K个三元组的限制下，找出1号点到N号点的最短路径，并输出任意一条合法路径，会有Check检查你的输出。

输入格式

输入文件第一行有三个数N，M，K，意义如题目所述。

接下来M行每行两个数A，B，表示A，B间有一条边。

再下面K行，每行三个数（A，B，C）描述一个三元组。

输出格式

输出文件共两行数，第一行一个数S表示最短路径长度。

第二行S+1个数，表示从1到N所经过的节点。

输入输出样例

4 4 2
1 2
2 3
3 4
1 3
1 2 3
1 3 4

输出

4
1 3 2 3 4

说明/提示

对于40%的数据满足N≤10，M≤20，K≤5。

对于100%的数据满足N≤3000，M≤20000，K≤100000。

思路：

通过分析题目发现边权为1容易想到用宽搜或者深搜来做，但是与一般的深搜和宽搜不同的是，本题是用边来搜索（不是一般的用点搜索）
因为要记录路径，所以用广搜找最短路，广搜的状态里加一个变量记录上一个点，最后dfs输出。

code：

#include <bits/stdc++.h>
#include <string.h>
#include <map>
#include <stack>
#include <queue>
using namespace std;
const int maxn=3e3+10;
struct node{
    int last,x,y,lev;
}nodes;
int g[maxn][maxn],pro[maxn][maxn],n,m,k;
int st[maxn][maxn];
bool f[maxn][maxn];
int s;
vector<int>res;
bool check(int l,int x,int y){
    if(g[x][y] && !f[x][y] && st[l][x]!=y ) return true;
    return false;
}
int bfs(){
    queue<node> q;
    for(int i=1;i<=n;i++){
        if(g[1][i]) {
            nodes.last=0,nodes.x=1,nodes.y=i,nodes.lev=1;
            f[nodes.x][nodes.y]=true;
            pro[nodes.y][nodes.x]=nodes.last;
            q.push(nodes);
        }
    }
    while(q.size()){
        node temp=q.front();
        q.pop();
        if(temp.y==n){
            s=temp.x;
            return temp.lev;
        }
        //cout<<temp.x<<" "<<temp.y<<endl;
        for(int i=1;i<=n;i++){
            if(check(temp.x,temp.y,i)){
                nodes.last=temp.x,nodes.x=temp.y,nodes.y=i,nodes.lev=temp.lev+1;
                f[nodes.x][nodes.y]=true;
                pro[nodes.y][nodes.x]=nodes.last;
                q.push(nodes);
            }
        }
    }
    return 0;
}
int main(){
    int x,y,w;
    cin>>n>>m>>k;
    for(int i=0;i<m;i++){
        scanf("%d%d",&x,&y);
        g[x][y]=1;
        g[y][x]=1;
    }
    for(int i=0;i<k;i++){
        scanf("%d%d%d",&x,&y,&w);
        st[x][y]=w;
    }
    cout<<bfs()<<endl;
    x=n,y=s;
    res.push_back(n);
    while(1){
        if(!y) break;
        res.push_back(y);
        int temp=x;
        x=y,y=pro[temp][y];
    }
    for(int i=res.size()-1;i>=0;i--) cout<<res[i]<<" ";
    system("pause");
    return 0;
}