POJ 1611 The Suspects

最新推荐文章于 2020-10-12 15:35:17 发布

yurio7r

最新推荐文章于 2020-10-12 15:35:17 发布

阅读量314

点赞数

文章标签：并查集

本文链接：https://blog.csdn.net/MrSiz/article/details/50963058

版权

Data Structures 专栏收录该内容

16 篇文章 0 订阅

订阅专栏

辅助数组cnt，统计节点数目。

#include <cstdio>
#include <iostream>

using namespace std;
const int MAX = 1e3 + 5;
int x[MAX], y[MAX], fa[MAX], flag[MAX], r[MAX];
int n, d;

void init(int n)
{
    for (int i = 1; i <= n; ++i)
    {
        fa[i] = i;
        flag[i] = 0;
        r[i] = 1;
    }
}
int find(int x)
{
    return fa[x] == x ? x : fa[x] = find(fa[x]);
}

void unite(int a, int b)
{
    int t_a = a;
    int t_b = b;
    a = find(a);
    b = find(b);
    if (a == b)
        return ;
    if ((x[t_a] - x[t_b]) * (x[t_a] - x[t_b]) + (y[t_a] - y[t_b]) * (y[t_a] - y[t_b]) <= d * d && r[a] < r[b])
    {
        fa[a] = b;
    }
    else if ((x[t_a] - x[t_b]) * (x[t_a] - x[t_b]) + (y[t_a] - y[t_b]) * (y[t_a] - y[t_b]) <= d * d && r[a] >= r[b])
    {
        fa[b] = a;
        if (r[b] == r[a])
            r[b]++;
    }

}

int main()
{
    scanf("%d%d", &n, &d);
    init(n);
    for (int i = 1; i <= n; ++i)
    {
        scanf("%d%d", &x[i], &y[i]);
    }
    char type;
    while (scanf(" %c", &type)!= EOF)
    {
        if (type == 'O')
        {
            int id;
            scanf("%d", &id);
            flag[id] = 1;
            for (int i = 1; i <= n; ++i)
            {
                if (flag[i] && i != id)
                {
                    unite(i, id);
                }
            }
        }
        else if (type == 'S')
        {
            int id1, id2;
            scanf("%d%d", &id1, &id2);
            if (find(id1) == find(id2))
                printf("SUCCESS\n");
            else
                printf("FALL\n");
        }
    }

    return 0;
}