60分求助
查看原帖
60分求助
500031
FJ_OIer楼主2022/8/24 10:17

思路是k次bfs遍历,遍历到一个点就把这个点能到达的奶牛数+1,最后的答案就是有几个点能到达k只奶牛。

#include <iostream>
#include <queue>
#include <vector>
using namespace std;
int k,n,m,d[101],ans,cow[1001];
vector<int> v[1001];
void bfs(int start)
{
    queue<int> q;
    bool vis[1001]={};
    vis[start]=1;
    q.push(start);
    while (q.size()){
        int now=q.front();
        q.pop();
        for (int i=0;i<v[now].size();i++){
            int to=v[now][i];
            if (!vis[to]){
                q.push(to);
                vis[to]=1;
                cow[to]++;
            }
        }
    }
}
int main()
{
    cin>>k>>n>>m;
    for (int i=1;i<=k;i++){
        cin>>d[i];
        cow[d[i]]=1;
    }
    for (int i=1;i<=m;i++){
        int s,e;
        cin>>s>>e;
        v[s].push_back(e);
    }
    for (int i=1;i<=k;i++){
        bfs(d[i]);
    }
    for (int i=1;i<=n;i++){
        if (cow[i]==k){
            ans++;
        }
    }
    cout<<ans;
    return 0;
}

2022/8/24 10:17
加载中...