bitset的运用

题目链接: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;
}
上一篇:日常Java 2021/10/14


下一篇:AT3857-[AGC020C]Median Sum【背包,bitset】