给定一张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
注意:int 存储30位,unsigned int 存储31位
② STL bitse效率更高
TLE代码
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
const int N=30010;
unsigned int t[N][1001];
int n,m,head[N],nex[N],k[N],deg[N],v[N],tot,ans[N];
int cha(unsigned int x)
{
int cnt=0;
while(x>0)
{
cnt++;
x=x&(x-1);
}
return cnt;
}
void topsort()
{
queue<int> q;
for (int i=1; i<=n; i++)if(deg[i]==0)q.push(i);
while(q.size())
{
int x=q.front();
q.pop();k[++tot]=x;
for (int i=head[x]; i; i=nex[i])
{
deg[v[i]]--;
if(deg[v[i]]==0)q.push(v[i]);
}
}
}
int main()
{
cin>>n>>m;
int x,y;
for (int i=1; i<=m; i++)
{
scanf("%d%d",&x,&y);
nex[i]=head[x];
head[x]=i;v[i]=y;
deg[y]++;
}
topsort();
for (int i=n; i>=1; i--)
{
t[k[i]][(k[i]-1)/32+1]|=1<<((k[i]-1)%32);
for (int j=head[k[i]]; j; j=nex[j])
{
for (int g=1; g<=n/32+1; g++)
{
t[k[i]][g]=(unsigned int)t[k[i]][g]|t[v[j]][g];
}
}
for (int j=1; j<=n/32+1; j++)ans[k[i]]+=cha(t[k[i]][j]);
}
for (int i=1; i<=n; i++)cout<<ans[i]<<endl;
return 0;
}
AC代码
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
#include<bitset>
using namespace std;
const int N=30010;
int n,m,head[N],nex[N],k[N],deg[N],v[N],tot,ans[N];
bitset<N> f[N];
void topsort()
{
queue<int> q;
for (int i=1; i<=n; i++)if(deg[i]==0)q.push(i);
while(q.size())
{
int x=q.front();
q.pop();k[++tot]=x;
for (int i=head[x]; i; i=nex[i])
{
deg[v[i]]--;
if(deg[v[i]]==0)q.push(v[i]);
}
}
}
int main()
{
cin>>n>>m;
int x,y;
for (int i=1; i<=m; i++)
{
scanf("%d%d",&x,&y);
nex[i]=head[x];
head[x]=i;v[i]=y;
deg[y]++;
}
topsort();
for (int i=n; i>=1; i--)
{
f[k[i]][k[i]]=1;
for (int j=head[k[i]]; j; j=nex[j])
{
f[k[i]]|=f[v[j]];
}
}
for (int i=1; i<=n; i++)cout<<f[i].count()<<endl;
return 0;
}