7-4 疫情防控(30分)【并查集】【2021 RoboCom 世界机器人开发者大赛-本科组（初赛）】

考点：并查集 + 思维（逆序操作）+ 邻接表记录优化

7-4 疫情防控 (30分)

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

为了研究疫情防控对于跨地区交通运输的影响，假设现在有 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

编译器

NO_COMPILER

代码

#include<bits/stdc++.h> 
using namespace std;

bool openn[400005];//第i个地点是开通的 
int n, m, d, a[400005], b[400005];//a b放的是地点 
int c[400005], q[400005], fa[400005], ans[1005];
int xx[1005][1005], yy[1005][1005];  //第i天第j次询问的地点的编号 
vector<int> nextt[50005];

int findd(int x){
	if(fa[x] == x) return x;
	else return fa[x] = findd(fa[x]);
}

int main(){
	cin>>n>>m>>d;//机场数、航线数以及新增防控地区的天数 
	for(int i = 1; i<=n; ++i) openn[i] = 1;
	for(int i = 1; i<=m; ++i){
		cin>>a[i]>>b[i];
		nextt[a[i]].push_back(b[i]);
		nextt[b[i]].push_back(a[i]);
	}
	
	for(int i = 1; i<=d; ++i){
		cin>>c[i]>>q[i];
		for(int j = 1; j<=q[i]; ++j)
			cin>>xx[i][j]>>yy[i][j];
		//close
		openn[c[i]] = 0;
	}
	
	for(int k = 1; k<=n; ++k) fa[k] = k;
	
	//先连一次啊，别漏了啊
	for(int i = 1; i<=m; ++i){
		if(openn[a[i]] && openn[b[i]]) fa[findd(a[i])] = findd(b[i]);
	} 
	
	for(int i = d; i>=1; --i){
		if(d!=i){
			//open && unite
			int p1 = c[i+1];
			openn[p1] = 1;  //open
			for(int j = 0; j<nextt[p1].size(); ++j){
				int p2 = nextt[p1][j];
//				cout<<"debug:"<<p1<<" "<<p2<<endl; 
				if( openn[ p2 ] ) 
					fa[findd(p1)] = findd(p2);
			}
		}

		//check
		int temp = 0;
		for(int j = 1; j<=q[i]; ++j){
			if( findd(xx[i][j]) != findd(yy[i][j]) ) temp++;
		}
		ans[i] = temp;
	}
	for(int i = 1; i<=d; ++i) cout<<ans[i]<<endl;

    return 0;
}

编译器输出

a.cpp: In function ‘int main()’:
a.cpp:48:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
    for(int j = 0; j<nextt[p1].size(); ++j){
                   ~^~~~~~~~~~~~~~~~~

测试点	结果	测试点得分	耗时	内存
0	答案正确	15	4.00 ms	1588 KB
1	答案正确	1	5.00 ms	1716 KB
2	答案正确	1	5.00 ms	1600 KB
3	答案正确	5	11.00 ms	2868 KB
4	答案正确	8	531.00 ms	15160 KB

评测结果 答案正确 (30 分)