L2-013 红色警报_pta_天梯赛_dfs解法

L2-013 红色警报

分数 25

战争中保持各个城市间的连通性非常重要。本题要求你编写一个报警程序，当失去一个城市导致国家被分裂为多个无法连通的区域时，就发出红色警报。注意：若该国本来就不完全连通，是分裂的k个区域，而失去一个城市并不改变其他城市之间的连通性，则不要发出警报。

输入格式：

输入在第一行给出两个整数N（0 < N ≤ 500）和M（≤ 5000），分别为城市个数（于是默认城市从0到N-1编号）和连接两城市的通路条数。随后M行，每行给出一条通路所连接的两个城市的编号，其间以1个空格分隔。在城市信息之后给出被攻占的信息，即一个正整数K和随后的K个被攻占的城市的编号。

注意：输入保证给出的被攻占的城市编号都是合法的且无重复，但并不保证给出的通路没有重复。

输出格式：

对每个被攻占的城市，如果它会改变整个国家的连通性，则输出Red Alert: City k is lost!，其中k是该城市的编号；否则只输出City k is lost.即可。如果该国失去了最后一个城市，则增加一行输出Game Over.

输入样例：

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

输出样例：

City 1 is lost.
City 2 is lost.
Red Alert: City 0 is lost!
City 4 is lost.
City 3 is lost.
Game Over.

思路:

(该题比较简单,测试范围也比较小,所以我用了比较的方法,虽然我也不会什么厉害的技巧)

1用一个二维数组city存入城市之间的关系

2每沦陷一个城市x,用一个数组v记录能和x联通的城市

3利用数组v写一个双重循环,每一次循环dfs一次检查是否联通,如果不联通则城市联通性被破坏

4dfs就能联通就一直往下找,直到两个参数x,y相等,说明联通

#include<iostream>
#include<vector>
#include<string.h>
using namespace std;
vector<int> v;
int city[510][510];
int visit[510];
int n;
int dfs(int x,int y)
{
	
	visit[x]=1;
//	cout<<x<<"   "<<y<<" f \n";
	if(x==y)
		return 1;
    for(int i=0;i<n;i++)
    {
        if(city[x][i] && !visit[i])
        {
            if(dfs(i,y))
                return 1;
        }
    }
    return 0;
}
int check()
{
    for(int i=0;i<v.size();i++)
    {
        for(int j=i+1;j<v.size();j++)
        {
            if(i!=j)
            {
            	//cout<<v[i]<<" "<<v[j]<<endl;
            	memset(visit,0,sizeof visit);
                if(!dfs(v[i],v[j]))
                {
                    return 0;//destory
                }
            }
        }
    }
    return 1;
}
int main()
{
    int m,d,x,y;
    cin>>n>>m;
    for(int i=0;i<m;i++)
    {
        cin>>x>>y;
        city[x][y]=city[y][x]=1;
    }
    int k;
    cin>>k;
    for(int i=0;i<k;i++)
    {
        cin>>d;
        for(int j=0;j<n;j++)
        {
            if(city[d][j])
            {
                city[d][j]=city[j][d]=0;
                v.push_back(j);
            }
        }
      //  cout<<endl;
        if(check())
        {
            printf("City %d is lost.\n",d);
        }
        else
        {
            printf("Red Alert: City %d is lost!\n",d);
        }
        v.clear();
    }
    if(k>=n)
    {
    	cout<<"Game Over.";
	}
}