城市相通问题

/*
给你n个城市,告诉你有m条路径是相通的,接下来有q次询问,每一次询问问你两个城市之间是否可以相互到达,如果可以YES,不可以输出NO

输入
输入第一行一个整数T(1<=T<=5),接下来有T组测试数据
对于每一组测试数据,第一行输入两个数n,m(1<=n,m<=100000)
代表有n个城市,m条路径
接下来m行,每一行两个数代表两个城市之间可以到达,城市标号为1到n
接下来输入一个数Q,代表Q次询问(1<=Q<=1000)
接下来Q行,每一行两个数,代表要查询的两个城市之间是否可以到达

输出
对于每一次询问,输出对应的答案

样例
输入
1
5 5
1 2
2 3
1 3
1 4
1 2
3
1 4
1 5
2 4

输出
YES
NO
YES
*/

#include<bits/stdc++.h>
using namespace std;
int City1,City2;
int flag;
int vis[100005];
vector<int>ve[100005];
void dfs(int FromThisCity)//从这个城市(结点)开始
{
    if(FromThisCity==City2)
    {
        flag=1;//如果dfs能遍历访问到这个城市(结点),说明可以到达,flag标志1
        return ;
    }
    for(int i=0;i<ve[FromThisCity].size();i++)//这个城市(结点)共有几个相连的城市(结点)
    {
        if(!vis[ ve[FromThisCity][i] ])//如果当前结点未访问过则访问,这里的 i 指相连的第i个城市
        {
            vis[ ve[FromThisCity][i] ]=1;//访问此结点,并标记为1
            dfs(ve[FromThisCity][i]);//从这个结点继续向下(深)访问
        }
    }
}

int main()
{
    int t;
    cin>>t;
    int n,m;
    while(t--)
    {
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)
            ve[i].clear();
        for(int i=1;i<=m;i++)
        {
            int ThisCity,ThatCity;
            scanf("%d%d",&ThisCity,&ThatCity);//这两个城市之间可以到达
            ve[ThisCity].push_back(ThatCity);
            ve[ThatCity].push_back(ThisCity);
        }
        int q;
        scanf("%d",&q);
        while(q--)
        {
            scanf("%d%d",&City1,&City2);
            flag=0;
            memset(vis,0,sizeof(vis));//判断数组vis全部置零
            vis[City1]=1;//第一个城市(结点)已访问
            dfs(City1);//从第一个城市(结点)开始dfs遍历
            if(flag)
                cout<<"YES"<<endl;
            else
                cout<<"NO"<<endl;
        }
    }

    return 0;
}

猜你喜欢

转载自blog.csdn.net/Jason6620/article/details/89220277