Codeforces 396 div.2 带权并查集_codeforces 带权并查集-优快云博客

本文链接：https://blog.youkuaiyun.com/tree__water/article/details/54986427

本文介绍了一种使用带权并查集的方法来处理单词间的关系查询问题。给定N个单词及M组已知的同义词或反义词关系，通过构建并查集的数据结构，可以高效地查询任意两个单词之间的关系（同义、反义或不确定）。文章详细解释了算法原理，并提供了完整的C++代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意：
一共给你N个单词，其中有M个关系是已知的，我们要对Q对单词进行关系查询。
其中M个关系，要么两个单词是同义词，要么两个单词是反义词。
对于查询，如果两个单词是同义词，输出1，如果是反义词输出2，如果不能确定输出3.
分析：
带权并查集。权值的设置需要考虑

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#include <string>
#include <map>
using namespace std;
typedef long long ll;
const int N=1e5+5;
int fa[N],ran[N];
int _find(int x)
{
    if(fa[x]!=x)
    {
        int pre=fa[x];
        fa[x]=_find(fa[x]);
        ran[x]=(ran[x]+ran[pre])%2;
       //a->fa>ffa
       //ran[a]是a和fa的关系，ran[pre]是fa和ffa的关系，新的ran[a]是a和ffa的关系，根据关系向量得该表达式
    }
    return fa[x];
}
int main()
{
    int n,m,q,r;
    char sr[25],ssr[25];
    scanf("%d%d%d",&n,&m,&q);
    map<string,int> mp;
    for(int i=1;i<=n;i++)
    {
        scanf("%s",sr);
        mp[sr]=i;
        fa[i]=i,ran[i]=0;
    }
    for(int i=0;i<m;i++)
    {
        scanf("%d%s%s",&r,sr,ssr);
        int a=mp[sr],b=mp[ssr];
        int aa=_find(a),bb=_find(b);
        if(aa!=bb)
        {
            puts("YES");
            fa[bb]=aa;
            ran[bb]=(ran[a]-ran[b]+r+1)%2;
        }
        else
        {
            n=(ran[a]-ran[b]+2)%2;
            n==r-1?puts("YES"):puts("NO");
        }
    }
    while(q--)
    {
        scanf("%s%s",sr,ssr);
        int a=mp[sr],b=mp[ssr];
        int aa=_find(a),bb=_find(b);
        if(aa!=bb)puts("3");
        else
        {
            n=(ran[a]-ran[b]+2)%2;
            printf("%d\n",n+1);
        }
    }
}