AcWing 164. 可达性统计(bitset、拓扑序)

2021-03-08 12:26:56 浏览数 (1)

思路:本质上还是一个动态规划 父亲点答案等于所有儿子数 1,暴力会超时,需要用bitset压位优化常数

注意bitset.count()方法是数一个串1的个数~

代码语言:javascript复制
#include<bits/stdc  .h>
using namespace std;
const int N=30010,M=30010;
int n,idx,e[M],ne[M],h[N],d[N],m;
queue<int>q;
vector<int>ans;
bitset<N>bit[N];
void add(int a,int b){
    e[idx]=b,ne[idx]=h[a],h[a]=idx  ;
}
void tp(){
    for(int i=1;i<=n;i  ){
        if(!d[i])q.push(i);
    }
    while(!q.empty()){
        int x=q.front();
        ans.push_back(x);
        q.pop();
        for(int i=h[x];~i;i=ne[i]){
            int j=e[i];
            if(--d[j]==0){
                q.push(j);
            }
        }
    }
    
    for(int i=ans.size()-1;i>=0;i--){
        int j=ans[i];
        bit[j][j]=1;
        for(int k=h[j];~k;k=ne[k]){
            bit[j]|=bit[e[k]];
        }
    }
}
int main(){
    cin>>n>>m;
    memset(h,-1,sizeof h);
    for(int i=1;i<=m;i  ){
        int a,b;
        cin>>a>>b;
        d[b]  ;
        add(a,b);
    }
    tp();
    for(int i=1;i<=n;i  ){
        cout<<bit[i].count()<<endl;
    }

}

0 人点赞