数据范围 1 <= N, M <= 30000
思路:
我们可以发现,父节点能经过的点,是它所有子节点能经过的点的并集。所以我们先统计子节点能经过的点,然后再根据子节点的并集得出父节点能经过的点个数。
因为题目保证没有环出现。于是,我们就可以先拓扑排序,将得到的序列从后向前遍历。那么就一定是先算的子节点,然后才能算到父节点。
bitset可以用来做二进制运算(安利一篇别人的文章,还不错:bitset简单应用)。其余细节看下面的代码注释。
代码:
#include<bits/stdc++.h>
using namespace std;
const int N = 30000+5;
#define int long long
typedef pair<int, int>PII;
const int Max = 0x3f3f3f3f3f3f3f3f;
const int Min = -0x3f3f3f3f3f3f3f3f;
int n, m, k;
int T;
vector<int>v[N]; //图
queue<int>q; //拓扑排序所用队列
int ru[N]; //拓扑排序,入度
int d[N]; //储存拓扑序列
int cnt = 0;
bitset<N>num[N];
signed main()
{
ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
cin >> n >> m;
while(m --)
{
int x, y;
cin >> x >> y;
v[x].push_back(y);
ru[y] ++;
}
//拓扑排序
for(int i = 1; i <= n; i ++)
if(ru[i]==0) q.push(i), d[cnt++] = i;
while(q.size())
{
int u = q.front();
q.pop();
for(int i = 0; i < v[u].size(); i ++)
{
int t = v[u][i];
ru[t] --;
if(ru[t]==0) q.push(t), d[cnt++] = t;
}
}
//初始化
for(int i = 1; i <= n; i ++) num[i][i] = 1; //自己能够到达自己,初始化成 1
//倒着遍历集合,合并
for(int i = cnt-1; i >= 0; i --)
{
int u = d[i];
for(auto t : v[u])
{
num[u] |= num[t]; //或,合并集合
}
}
//输出
for(int i = 1; i <= n; i ++)
{
cout << num[i].count(); //统计二进制中一的个数
cout << "\n";
}
}