【HDU】畅通工程

Problem Description

某省调查城镇交通状况，得到现有城镇道路统计表，表中列出了每条道路直接连通的城镇。省政府“畅通工程”的目标是使全省任何两个城镇间都可以实现交通（但不一定有直接的道路相连，只要互相间接通过道路可达即可）。问最少还需要建设多少条道路？ 
 

Input

测试输入包含若干测试用例。每个测试用例的第1行给出两个正整数，分别是城镇数目N ( < 1000 )和道路数目M；随后的M行对应M条道路，每行给出一对正整数，分别是该条道路直接连通的两个城镇的编号。为简单起见，城镇从1到N编号。 注意:两个城市之间可以有多条道路相通,也就是说3 31 21 22 1这种输入也是合法的当N为0时，输入结束，该用例不被处理。 
 

Output

对每个测试用例，在1行里输出最少还需要建设的道路数目。 
 

Sample Input

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

Sample Output

1
0
2
998

Solution

#include <iostream>
#include <memory.h>
using namespace std;

int n, m;
int pre[1000];
int answer;

int find(int w){
    int r = w;
    while(pre[r] != r){
        r = pre[r];
    }
    int i = w;
    int j;
    while(i != r){
        j = pre[i];
        pre[i] = r;
        i = j;
    }
    return r;
}

void join(int s, int d){
    int p1 = find(s);
    int p2 = find(d);
    if(p1 != p2){
        pre[p2] = p1;
        answer--;
    }
}

int main(){

    while(cin >> n >> m){
        if(n == 0)
            break;

        answer = n-1;
        for(int i = 1; i <= n; i++){
            pre[i] = i;
        }
        for(int i = 0; i < m; i++){
            int s,d;
            cin >> s >> d;
            join(s, d);
        }
        cout << answer << endl;
    }

    return 0;
}