AcWing 3250. 通信网络 两次bfs

AcWing 3250. 通信网络

某国的军队由 N 个部门组成,为了提高安全性,部门之间建立了 M 条通路,每条通路只能单向传递信息,即一条从部门 a 到部门 b 的通路只能由 a 向 b 传递信息。

信息可以通过中转的方式进行传递,即如果 a 能将信息传递到 b,b 又能将信息传递到 c,则 a 能将信息传递到 c。

一条信息可能通过多次中转最终到达目的地。

由于保密工作做得很好,并不是所有部门之间都互相知道彼此的存在。

只有当两个部门之间可以直接或间接传递信息时,他们才彼此知道对方的存在。

部门之间不会把自己知道哪些部门告诉其他部门。

AcWing 3250. 通信网络 两次bfs

上图中给了一个 4 个部门的例子,图中的单向边表示通路。

部门 1 可以将消息发送给所有部门,部门 4 可以接收所有部门的消息,所以部门 1 和部门 4 知道所有其他部门的存在。

部门 2 和部门 3 之间没有任何方式可以发送消息,所以部门 2 和部门 3 互相不知道彼此的存在。

现在请问,有多少个部门知道所有 N 个部门的存在。

或者说,有多少个部门所知道的部门数量(包括自己)正好是 N。

输入格式
输入的第一行包含两个整数 N,M,分别表示部门的数量和单向通路的数量。所有部门从 1 到 N 标号。

接下来 M 行,每行两个整数 a,b,表示部门 a 到部门 b 有一条单向通路。

输出格式
输出一行,包含一个整数,表示答案。

数据范围
对于 30% 的评测用例,1≤N≤10,1≤M≤20;
对于 60% 的评测用例,1≤N≤100,1≤M≤1000;
对于 100% 的评测用例,1≤N≤1000,1≤M≤10000。

输入样例:
4 4
1 2
1 3
2 4
3 4
输出样例:
2

样例解释
部门 1 和部门 4 知道所有其他部门的存在。

这道题我还在想怎么遍历比较好,y总的视频却给了我一个思路,一个点正着进行一次,反着进行一次,然后看看有哪些点标记了,不愧是y总,太强了,

代码如下

#include<iostream>
#include<algorithm>
#include<cstring>

using namespace std;

const int N=1e3+10,M=2e4+10;

int h1[N],h2[N],ne[M],e[M],idx;
int n,m;
bool st1[N],st2[N];

void add(int h[],int a,int b)
{
    e[idx]=b;ne[idx]=h[a];h[a]=idx++;
}

void dfs(int h[],int x,bool st[])
{
    st[x]=true;
    for(int i=h[x];~i;i=ne[i])
    {
        int j=e[i];
        if(!st[j])
        dfs(h,j,st);
    }
}

int main(void)
{
    memset(h1,-1,sizeof h1);
    memset(h2,-1,sizeof h2);
    cin>>n>>m;
    int ans=0;
    while(m--)
    {
        int a,b;
        cin>>a>>b;
        add(h1,a,b);
        add(h2,b,a);
    }
    for(int i=1;i<=n;i++)
    {
        memset(st1,0,sizeof st1);
        dfs(h1,i,st1);
        memset(st2,0,sizeof st2);
        dfs(h2,i,st2);
        int res=0;
        for(int j=1;j<=n;j++)
        if(st1[j]||st2[j])
        res++;
        if(res==n)
        ans++;
    }
    cout<<ans;
}
上一篇:线性筛质数


下一篇:剑指 Offer 03. 数组中重复的数字