暑假前专题题解---图论---A

        对于这个题,我们建立反向边即可,然后从k开始bfs,第一个入队的,出度为0的点,到这个点的距离就是我们要的答案。

注意原点的权重初始值是1 ,自己本身也是一门要选的课。

代码如下

#include <bits/stdc++.h>
using namespace std;
typedef pair<int, int> pi;
const int maxn = 2e5 + 10;
vector<int> G[maxn];
int vis[maxn],ans = 0;
void bfs(int s){
    queue<pair<int, int> > q;
    q.push(make_pair(s, 1));
    while (!q.empty()) {
        pi now = q.front(); q.pop();
        int dis = now.second,u = now.first;
        if (G[u].size() == 0) { ans = dis; return; }
        if (vis[u]) continue;
        vis[u] = 1;
        for (int i=0; i<G[u].size(); i++) {
            int v = G[u][i];
            if (!vis[v]){
                q.push(make_pair(v, dis+1));
            }
        }
    }
}
int main(){
    int n,k,m,u,v;
    scanf("%d%d%d",&n,&k,&m);
    for (int i=0; i<m; i++) {
        scanf("%d%d",&u,&v);
        G[v].push_back(u);
    }
    bfs(k);
    cout << ans << endl;
    return 0;
}

猜你喜欢

转载自blog.csdn.net/CCCCTong/article/details/81487125