题目链接:https://ac.nowcoder.com/acm/problem/51011
- 说一下第二种拓扑排序的思路,因为是有向无环图,可以进行拓扑排序,记录出队列顺序, 再倒着,由结尾统计。
- 运用 二维bitset 的 | 运算操作可以 顺带着去重。
代码如下:
第一种 (直接dfs 203ms)
#include <bits/stdc++.h>
using namespace std;
#define rep(x, y, z) for (int x = y; x <= z; x++)
#define dec(x, y, z) for (int x = y; x >= z; x--)
const int N = 30010, mod = 1e9 + 7;
const double expp = 1e-10;
typedef long long ll;
//------------------------'
int n,m;
int h[N],e[N],ne[N],idx;
bitset<N> f[N];
int st[N];
void add(int a,int b) {
e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
void dfs(int u){
f[u][u] = 1;
for(int i=h[u];~i;i=ne[i]){
int j = e[i];
if(!f[j].any())
dfs(j);
f[u] |= f[j];
}
}
int main() {
memset(h,-1,sizeof h);
cin>>n>>m;
rep(i,1,m) {
int a,b;
cin>>a>>b;
add(a,b);
}
rep(i,1,n) {
if(!f[i].any())
dfs(i);
}
rep(i,1,n) cout<<f[i].count()<<endl;
return 0;
}
第二种(拓扑排序 151ms)
#include <bits/stdc++.h>
using namespace std;
#define rep(x, y, z) for (int x = y; x <= z; x++)
#define dec(x, y, z) for (int x = y; x >= z; x--)
const int N = 30010, mod = 1e9 + 7;
const double expp = 1e-10;
typedef long long ll;
//------------------------'
int n,m;
int h[N],e[N],ne[N],idx;
bitset<N> f[N];
int d[N];
vector<int> ans;
void add(int a,int b) {
f[a][b] = 1;
e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
void tour(){
queue<int> q;
rep(i,1,n) if(!d[i]) q.push(i);
while(q.size()){
auto t = q.front();
q.pop();
ans.push_back(t);
for(int i=h[t];~i;i=ne[i]){
int j = e[i];
if(!--d[j]) q.push(j);
}
}
}
int main() {
memset(h,-1,sizeof h);
cin>>n>>m;
rep(i,1,m) {
int a,b;
cin>>a>>b;
add(a,b);
d[b] ++;
}
tour();
for(int i=ans.size()-1;i>=0;i--){
int t = ans[i];
f[t][t] = 1;
for(int j=h[t];~j;j=ne[j]){
f[t] |= f[e[j]];
}
}
rep(i,1,n) cout<<f[i].count()<<endl;
return 0;
}