164. 可达性统计拓扑排序 / bitset

Posted 2022-03-19 辉小歌

无环说明一定有拓扑排序。

我们得到拓扑排序，倒着推。注意压位。

#include<bits/stdc++.h>
#include<bitset>
using namespace std;
const int N=30010;
int h[N],e[N],ne[N],idx;
int d[N],n,m;
vector<int>ve;
bitset<N>cnt[N]; //每一个点压了N位。1表示可以到达,0表示不可以到达
void add(int a,int b)

    e[idx]=b,ne[idx]=h[a],h[a]=idx++;

void top_sort()

    queue<int>q;
    for(int i=1;i<=n;i++) if(!d[i]) q.push(i);
    while(q.size())
    
        int u=q.front(); q.pop();
        ve.push_back(u);
        for(int i=h[u];i!=-1;i=ne[i])
        
            int j=e[i];
            if(--d[j]==0) q.push(j);
        
    

int main(void)

    memset(h,-1,sizeof h);
    cin>>n>>m;
    for(int i=0;i<m;i++)
    
        int a,b; cin>>a>>b;
        d[b]++;
        add(a,b);
    
    top_sort();
    for(int i=ve.size()-1;i>=0;i--)
    
        int u=ve[i];
        cnt[u][u]=1;//u到u是有一个
        for(int j=h[u];j!=-1;j=ne[j])
        
            int k=e[j];
            cnt[u]|=cnt[k];
        
    
    for(int i=1;i<=n;i++) cout<<cnt[i].count()<<endl;//统计1的个数
    return 0;