【题解】CH2101可达性统计 拓扑排序+状态压缩+bitset

题目链接

描述

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

输入格式

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

输出格式

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

样例输入

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

---

从x出发能够到达的点，是从“x的各个后继节点y”出发能够到达的点的并集再加上x自身。DAG上跑拓扑排序，反着进行状态压缩（bitset实现），最后答案就是1的个数

#include<cstdio>
#include<queue>
#include<bitset>
#include<cstring>
using namespace std;
const int N=3e4+10;
bitset<N>f[N];
queue<int>q;
int head[N],tot,n,m,deg[N],tps[N],tp;
struct Edge{
	int v,nx;
}e[N];
inline void addedge(int u,int v)
{
	e[tot].v=v;
	e[tot].nx=head[u];
	head[u]=tot++;
	deg[v]++;
}
void toposort()
{
	for(int i=1;i<=n;i++)if(!deg[i])q.push(i);
	while(!q.empty())
	{
		int u=q.front();q.pop();tps[++tp]=u;
		for(int i=head[u];~i;i=e[i].nx)
		{
			int v=e[i].v;
			if(--deg[v]==0)q.push(v);
		}
	}
}
int main()
{
	//freopen("in.txt","r",stdin);
	memset(head,-1,sizeof(head));
    scanf("%d%d",&n,&m);
    int u,v;
    for(int i=1;i<=m;i++)scanf("%d%d",&u,&v),addedge(u,v);
    toposort();
    for(int i=tp;i;i--)
    {
    	u=tps[i];
    	f[u][u]=1;
    	for(int j=head[u];~j;j=e[j].nx)
    	{
    		v=e[j].v;
    		f[u]|=f[v];
		}
	}
	for(int i=1;i<=n;i++)printf("%d\n",f[i].count());
	return 0;
}

总结

将拓扑排序和状态压缩结合，好题