PAT 1013. Battle Over Cities (25)

题目：1013. Battle Over Cities (25)

思路：1. 利用并查集，剔除掉选中城市的道路，构建无向联通图。通过代表元的个数来计算需要搭建的道路数。

            2.利用深度优先搜索。

代码：并查集

#include<iostream>#include<vector>using namespace std;struct road{int x,y;};//初始化根节点void init(int *c,int n){int i;for(i=1;i<=n;++i)c[i]=i;}//查找根节点int findroot(int *c,int j){if(c[j]==j)return j;else{int t=findroot(c,c[j]);c[j]=t;return t;                //return findroot(c,c[j]);//仅用上面这一条，会出现同一个连接图，根节点不一样的情况}}//构建连通图void makeset(int *c,int k,vector<road> r){int i,a,b;for(i=0;i<r.size();++i){if(r[i].x!=k && r[i].y!=k)//避免选中城市加入连通图{a=findroot(c,r[i].x);b=findroot(c,r[i].y);if(a!=b)c[a]=b;    //根节点改变}}}int main(){//输入int N,M,K;cin>>N>>M>>K;int i,n1,n2,n;vector<road> R(M);//输入城市联通信息，并加以排序//避免出现由于先出现编号大的城市导致根节点不一致的情况。for(i=0;i<M;++i){cin>>n1>>n2;R[i].x=n1;R[i].y=n2;}vector<int> q(K);for(i=0;i<K;++i)cin>>q[i];//查找int j,k,num;int city[1000];for(i=0;i<K;++i){init(city,N);            //初始化节点makeset(city,q[i],R);num=0;                for(j=1;j<=N;++j){if(city[j]==j)num++;}cout<<num-2<<endl; //-2，一是减去被选中的城市，而是num个独立城市之间需要num-1条道路}system("pause");return 0;}