PAT-A 1013 Battle Over Cities

题目大意

城市之间通过道路相互连接，然而在战争中，当一个城市被敌人破坏以后，与其相关的所有道路都会不可用，此时为了恢复交通，需要新建道路，求需要新建的道路的数目。

题目给出了一个图，图的结点是各个城市，图的边是城市间的道路，当一个节点不可用时，求需要新增多少条边才能让城市之间重新连通。

解题思路

第一反应是用查并集来做，对于查并集还不了解的朋友可以去看这篇博客，讲的还挺详细的：https://blog.csdn.net/Hacker_ZhiDian/article/details/60965556?utm_source=blogxgwz5

但是用在这一题上，写出来的查并集算法在最后一点会显示超时，我做了一些优化，包括把cin、cout改成scanf、printf，或者如上面的博客说的优化方法（在合并时添加对子集的深度的判断），都无济于事，如果有用查并集成功AC的大神欢迎留言交流。

最后迫于无奈，改成了dfs算法(bfs也能做)，原理是每次有城市节点不可用的时候，遍历一遍图来判断其有几个分支就ok了。

注意要点

AC代码里面的算法需要使用scanf、printf，若用cin、cout还是会有一个点超时。

我会在最后贴上未能AC的查并集算法，欢迎交流

AC代码(dfs)：

#include <cstdio>
#include <vector>
#include <cstring>
using namespace std;

void dfs(vector<vector<int>> &matrix,bool *vis,int st)
{
	vis[st]=true;
	for(auto iter=matrix[st].begin();iter!=matrix[st].end();++iter)
	{
		if(!vis[*iter])
			dfs(matrix,vis,*iter);
	}
}

int main()
{
	int num_cities,num_highways,num_query;
	scanf("%d%d%d",&num_cities,&num_highways,&num_query);
	vector<vector<int>> matrix(num_cities+1);
	for(int i=0;i<num_highways;++i)
	{
		int st,end;
		scanf("%d%d",&st,&end);
		matrix[st].push_back(end);
		matrix[end].push_back(st);
	}
	
	//dfs求解
	for(int i=0;i<num_query;++i)
	{
		int temp;
		scanf("%d",&temp);
		bool *vis=new bool[num_cities+1];
		memset(vis,false,sizeof(bool)*(num_cities+1));
		vis[temp]=true;
		int cnt=0;
		for(int i=1;i<=num_cities;++i)
		{
			if(!vis[i])
			{
				dfs(matrix,vis,i);
				++cnt;
			}
		}
		printf("%d\n",cnt-1);

		delete[] vis;
	}

	return 0;
}

未AC代码(查并集):

#include <cstdio>
using namespace std;

struct road
{
	int from;
	int to;
};

int find(int *num,int n)
{
	while(num[n]!=n)
		n=num[n];
	return n;
}

void merge(int *num,int n1,int n2)
{
	num[find(num,n1)]=find(num,n2);
}

int main()
{
	int num_cities,num_highways,num_query;
	scanf("%d%d%d",&num_cities,&num_highways,&num_query);
	road *r=new road[num_highways];
	for(int i=0;i<num_highways;++i)
		scanf("%d%d",&r[i].from,&r[i].to);
	int *cities=new int[num_cities+1];
	
	for(int i=0;i<num_query;++i)
	{
		int temp;
		scanf("%d",&temp);
		//查并集求解
		for(int i=1;i<=num_cities;++i)
			cities[i]=i;
		//将有道路联系的合并
		for(int i=0;i<num_highways;++i)
		{
			if(r[i].from!=temp&&r[i].to!=temp)
				merge(cities,r[i].from,r[i].to);
		}
		//遍历祖先数目
		int cnt=0;
		for(int i=1;i<=num_cities;++i)
		{
			if(cities[i]==i)
				++cnt;
		}
		printf("%d\n",cnt-2);
	}

	delete[] r,cities;
	return 0;
}