2021 RoboCom 世界机器人开发者大赛-本科组(初赛)7-4 疫情防控 (30 分)

该博客探讨了如何在疫情防控期间,针对中高风险地区机场的封闭,利用并查集数据结构来处理每日新增防控地区对跨地区航班的影响。题目给出了一种输入格式和输出格式,并提供了一段示例输入和输出。解决方案是通过离线处理所有操作,先删除所有需要封闭的机场,然后逆序恢复并处理查询,以计算受影响的航班数量。整个算法的时间复杂度为O(n+mlogn+qlogn)。
摘要由CSDN通过智能技术生成

疫情尚未结束,严防疫情反复。为了做好疫情防控工作,国内设置了地区风险等级,对于中高风险地区的人员采取限制移动、居家隔离等手段。

为了研究疫情防控对于跨地区交通运输的影响,假设现在有 N 个机场,M 条航线,每天都会新增一个防控地区,一个防控地区会导致一个机场无法正常运作,航线也自然无法正常运行,每天会有 Qi​ 对旅客从 Xi​ 机场前往 Yi​ 机场,请计算有多少对旅客会受到影响无法完成行程。

旅客只要能直达或通过若干次中转,且乘坐的所有航线的出发和到达机场都正常运作,即视作可完成行程。

输入格式:

输入第一行是三个整数 N,M,D (1≤N≤5×104, 1≤M≤2×105, 1≤D≤103), 表示机场数、航线数以及新增防控地区的天数。

接下来首先有 M 行,每行给出空格分隔的两个数字 A 和 B,表示编号为 A 和 B 的机场之间有一条航线。航线是双向的,机场编号从 1 到 N。

然后是 D 块输入,每块输入内第一行为空格分隔的两个整数 C 和 Q (1≤Q≤103),表示新增机场编号为 C 所在的城市为防控地区,今天有 Q 段行程。数据保证新增的城市之前一定不是防控地区。

接下来的 Q 行,每行是空格分隔的两个数字 X 和 Y,表示编号为 X 和 Y 的机场的一段行程。行程有可能包括之前就已经成为防控地区的城市。

输出格式:

对于每天的询问,请在一行中输出在新增了一个防控地区后当天的行程有多少不能成行。

输入样例:

5 5 3
1 2
1 3
1 5
2 5
3 4
4 3
1 3
1 4
2 3
5 3
3 4
2 3
3 5
1 3
2 3
2 5
3 4

输出样例:

1
2
3

 

思路:

归纳题意就是给出一张图,支持两种操作:

1) 删除一个点。

2) 询问两个点是否连通。

观察到无法简单地维护“删除”这个操作,考虑将所有操作离线,逆序处理。即,先删除要删除的所有点,再倒着往里加点,顺便把询问处理掉,用并查集维护即可。

时间复杂度为:O(n+mlogn+qlogn)。

#include<iostream>
#include<stack>
#include<vector>
using namespace std;
const int N = 50010, M = 200010, Q = 1010;
int n,m,d;
int f[N];
bool book[N];//点是否为已删除状态
struct Edge{
	int u,v;
};
vector<int> g[N];//邻接表 
vector<Edge> query;//询问离线化 
vector<int> alg,erap;//alg某个询问的个数 era删除的点 
stack<int> res;//结果 
int getf(int v){
	if(f[v] == v) return v;
	else return f[v] = getf(f[v]);
}
void merge(int a,int b){
	int t1 = getf(a);
	int t2 = getf(b);
	if(t1 != t2){
		f[t2] = t1;
	}
}
bool ismerge(int a,int b){
	return getf(a) == getf(b);
}
int main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	int n,m,d;
	cin>>n>>m>>d;
	for(int i = 1; i <= n; i++) f[i] = i;
	//离线操作 
	for(int i = 0 ; i < m; i++){
		int u,v;
		cin>>u>>v;
		g[v].push_back(u);
		g[u].push_back(v);
	}
	for(int i = 0; i < d; i++){
		int c,q;
		cin>>c>>q;
		book[c] = true;
		erap.push_back(c); 
		alg.push_back(q);
		for(int j = 0; j < q; j++){
			int u,v;
			cin>>u>>v;
			query.push_back({u,v});
		}
	}
	//将没被删除的相关点生成并查集
	for(int i = 1; i <= n; i++){ 
		if(!book[i]){
			for(auto p:g[i]){
				if(!book[p]) merge(i,p);
			}
		}
	}
	
	for(int i = alg.size()-1; i >= 0; i--){
		int count = 0;
		for(int j = 0; j < alg[i]; j++){
			Edge tmp = query[query.size()-1];
			if(!ismerge(tmp.u,tmp.v)){
				count++;
			}
			if(query.size()) query.pop_back();
		}
		res.push(count);
		//插入点 
		int p = erap[i];
		book[p] = false;
		for(int x:g[p]){
			if(!book[x]) merge(x,p);
		}
	} 
	while(res.size()){
		cout<<res.top(); res.pop();
		if(res.size()) cout<<'\n';
	} 
	return 0;
}

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值