HackerRank - journey-to-the-moon 【并查集】

最新推荐文章于 2024-03-17 09:45:17 发布

原创最新推荐文章于 2024-03-17 09:45:17 发布 · 314 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#并查集

并查集专栏收录该内容

8 篇文章

订阅专栏

本文介绍了一道HackerRank上的题目——journey-to-the-moon，通过使用并查集解决来自不同国家宇航员的最大配对问题。文章提供了完整的AC代码及解题思路。

HackerRank - journey-to-the-moon 【并查集】

这里写图片描述

题意
有很多不同的宇航员，有些宇航员来自同一个国家，有些宇航员来自不同的国家，然后美国航天局想要选出两名来自不同国家的宇航员，求出最大的选法。然后数据一对一对的给出给出的说明这两人是来自同一个国家的。

思路
使用并查集并压缩路径，然后最后在MAP 里面存放的就是有几个“祖先”，并且这个祖先里面下属加上它一共有多少人

然后求不同的人数就是排列组合

AC代码

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <deque>
#include <vector>
#include <queue>
#include <string>
#include <cstring>
#include <map>
#include <stack>
#include <set>
#include <cstdlib>
#include <ctype.h>
#include <numeric>
#include <sstream>
using namespace std;
typedef long long LL;
const double PI = 3.14159265358979323846264338327;
const double E  = 2.718281828459;  
const int MAXN = 0x3f3f3f3f;
const int MINN = 0xc0c0c0c0;
const int maxn = 1e5 + 5;
int pre[maxn];
int find(int x)
{
    int r = x;
    while (pre[r] != r)
        r = pre[r];
    pre[x] = r;                 
    return r; 
}
void join(int x, int y)
{
    int fx = find(x), fy = find(y);
    if (x != fy)
        pre[fx] = fy;
}   
int main()
{
    int n, m;
    int a, b;
    int i;
    map <int, int> q;
    q.clear();
    scanf("%d %d", &n, &m);
    for (i = 0; i < n; i++)
        pre[i] = i;
    for (i = 0; i < m; i++)
    {
        scanf("%d%d", &a, &b);
        join(a, b);
    }
    LL tot = 0;
    LL sum = 0;
    for (i = 0; i < n; i++)
    {
        q[find(i)]  ++;
        sum ++;
    }   
    map <int, int>::iterator it;
    for (it = q.begin(); it != q.end(); it++)
    {
        sum -= it -> second;
        tot += (it -> second) * sum;
    }
    printf("%lld\n", tot);
}