poj2236(并查集)

http://poj.org/problem?id=2236（poj2236 点击打开链接）

思路：先预处理初所有点到其他点的距离，输入’O’的时候更新并查集。

分析：最开始想要先把所有点之间的距离打表，但是不知道之后怎么和并查集联系，就是不知道怎么更新并查集，还是看了别人的思路后写的，进步的是没有看代码，代码完全是自己敲的，开心o(￣▽￣)ブ

代码：

#include <iostream>#include <string>#include <cstring>#include <algorithm>#include <cmath>#include <vector>#include <cstdio>#include <map>#include <cstdlib>#include <deque>#include <iterator>#define pi 2*acos(0)using namespace std;const int maxn = 1001 + 10;struct node{    int x,y;} nodes[maxn];int father[maxn];double dis[maxn][maxn];bool isok[maxn];int n,d;void init(){    for(int i = 0; i <= n; i++)    {        father[i] = i;    }    for(int i = 1; i < n; i++)    {        for(int j = i + 1; j <= n; j++)        {            dis[i][j] = sqrt((double)(nodes[i].x-nodes[j].x) * (nodes[i].x-nodes[j].x) + (double)(nodes[i].y-nodes[j].y) * (nodes[i].y-nodes[j].y));            dis[j][i] = sqrt((double)(nodes[i].x-nodes[j].x) * (nodes[i].x-nodes[j].x) + (double)(nodes[i].y-nodes[j].y) * (nodes[i].y-nodes[j].y));        }    }    memset(isok,false,sizeof(isok));//初始化所有的电脑都是坏的。}int find(int x){    if(x == father[x])        return x;    int y = find(father[x]);    return father[x] = y;}void unin(int a,int b){    int fa = find(a);    int fb = find(b);    father[fa] = fb;}int main(){    scanf("%d%d",&n,&d);        int a,b;        for(int i = 1; i <= n; i++)        {            scanf("%d%d%*c",&nodes[i].x,&nodes[i].y);        }        init();        int num1,num2;        char ch;        while(~scanf("%c",&ch))        {            if(ch == 'O')            {                scanf("%d%*c",&num1);                isok[num1] = true;                //每次修好一台电脑，就遍历所有的电脑，根据距离判断是否可以进行联系，可以的合并两颗树。                for(int i = 1; i <= n; i++)                {                    if(isok[i] && i != num1)                    {                        if((dis[num1][i] <= (double)d) || (dis[i][num1] <= (double)d) )                        {                            if(find(num1) != find(i))                            {                                unin(num1,i);                            }                        }                    }                }            }            else if(ch == 'S')            {                scanf("%d%d%*c",&num1,&num2);                if(find(num1) == find(num2))//查询，如果根节点相同，就可以联系。                    printf("SUCCESS\n");                else                    printf("FAIL\n");            }        }}

• 有时候坚持一下下，别去看代码，万一自己写出来了呢，是吧。