【AcWing 164】 可达性统计 拓扑排序 bitset优化

给定一张N个点M条边的有向无环图，分别统计从每个点出发能够到达的点的数量。

输入格式
第一行两个整数N,M，接下来M行每行两个整数x,y，表示从x到y的一条有向边。

输出格式
输出共N行，表示每个点能够到达的点的数量。

数据范围
1≤N,M≤30000
输入样例：
10 10
3 8
2 3
2 5
5 9
5 9
2 3
3 9
4 8
2 10
4 9
输出样例：
1
6
3
3
2
1
1
1
1
1

题意：如题

思路：

如果想到点了这个题就不是什么难题。首先，这是一张有向无环图，那么我们可以先拓扑一遍，把点的遍历顺序存储下来。然后从末端结点开始倒过来处理，当前结点的结果就是其所有子节点的结果的并集，而因为是倒过来处理的，子节点的结果已经计算好了，所以就只需要将子节点结果简单合并即可。这里用bitset数组优化

AC代码：

#include<iostream>
#include<string>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<map>
#include <queue>
#include<sstream>
#include <stack>
#include <set>
#include <bitset>
#include<vector>
#include <unordered_set>
#define FAST ios::sync_with_stdio(false)
#define abs(a) ((a)>=0?(a):-(a))
#define sz(x) ((int)(x).size())
#define all(x) (x).begin(),(x).end()
#define mem(a,b) memset(a,b,sizeof(a))
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define rep(i,a,n) for(int i=a;i<=n;++i)
#define per(i,n,a) for(int i=n;i>=a;--i)
#define pb push_back
#define mp make_pair
#define fi first
#define se second
using namespace std;
typedef long long ll;
typedef pair<ll,ll> PII;
const int maxn = 3e4+2;
const int inf=0x3f3f3f3f;
const double eps = 1e-7;
const double pi=acos(-1.0);
const int mod = 1e9+7;
inline int lowbit(int x){return x&(-x);}
ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
void ex_gcd(ll a,ll b,ll &d,ll &x,ll &y){if(!b){d=a,x=1,y=0;}else{ex_gcd(b,a%b,d,y,x);y-=x*(a/b);}}//x=(x%(b/d)+(b/d))%(b/d);
inline ll qpow(ll a,ll b,ll MOD=mod){ll res=1;a%=MOD;while(b>0){if(b&1)res=res*a%MOD;a=a*a%MOD;b>>=1;}return res;}
inline ll inv(ll x,ll p){return qpow(x,p-2,p);}
inline ll Jos(ll n,ll k,ll s=1){ll res=0;rep(i,1,n+1) res=(res+k)%i;return (res+s)%n;}
inline ll read(){ ll f = 1; ll x = 0;char ch = getchar();while(ch>'9'||ch<'0') {if(ch=='-') f=-1; ch = getchar();}while(ch>='0'&&ch<='9') x = (x<<3) + (x<<1) + ch - '0',  ch = getchar();return x*f; }
int dir[4][2] = { {1,0}, {-1,0},{0,1},{0,-1} };

ll head[maxn],  pnt[maxn],  nxt[maxn] ,in[maxn] ,e = 0;
ll n,m; vector<ll> ans;
bitset<maxn> f[maxn];

void add(ll x, ll y)
{
    pnt[e] = y;
    nxt[e] = head[x];
    head[x] = e++;
}

void Topsort()
{
    queue<ll> q;
    rep(i,1,n) if(in[i]==0) q.push(i);
    while(!q.empty())
    {
        ll x = q.front();
        q.pop();
        ans.pb(x);
        for(ll i = head[x]; i!=-1; i = nxt[i])
        {
            ll v = pnt[i]; in[v]--;
            if(in[v]==0) q.push(v);
        }
    }
}

void solve()
{
    for(int i=ans.size()-1;i>=0;i--)
    {
        ll x = ans[i];
        f[x][x] = 1;
        for(ll i = head[x]; i!=-1; i=nxt[i])
        {
            ll v = pnt[i];
            f[x] |= f[v];
        }
    }
}

int main()
{
    cin>>n>>m;
    mem(head,-1);
    rep(i,1,m)
    {
        ll x, y;
        cin>>x>>y;
        in[y] ++;
        add(x,y);
    }
    Topsort();
    solve();
    rep(i,1,n)
    cout<<f[i].count()<<endl;
    return 0;
}