莫欺少年穷,修魔之旅在这开始—>算法提高课题解
思路:
1. 计算一遍拓扑排序
2. 逆序从后往前用位运算的方式计算能到达的点数
拓扑排序可参考: 有向图的拓扑序列
#include<bits/stdc++.h>
using namespace std;
const int N = 30010;
int n,m;
int h[N],e[N],ne[N],idx;
int d[N];
int q[N];
bitset<N> f[N];
void add(int a,int b)
{
e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
void topsort()
{
int hh=0,tt=-1;
for(int i=1;i<=n;i++)
if(!d[i])
q[++tt]=i;
while(hh<=tt)
{
int t=q[hh++];
for(int i=h[t];~i;i=ne[i])
{
int j=e[i];
if(--d[j]==0) q[++tt]=j;
}
}
}
int main()
{
cin>>n>>m;
memset(h,-1,sizeof h);
while(m--)
{
int a,b;
cin>>a>>b;
add(a,b);
d[b]++;
}
topsort();
//用位运算的方式计算集合
for(int i=n-1;i>=0;i--)
{
int j=q[i];
//自己肯定能够到达自己
f[j][j]=1;
//枚举这个点能够到达的点
for(int k=h[j];~k;k=ne[k])
f[j]|=f[e[k]];
}
for(int i=1;i<=n;i++) cout<<f[i].count()<<endl;
return 0;
}