Codeforces 396 div.2 带权并查集

来源：互联网发布：766冒险岛数据库进不了编辑：程序博客网时间：2024/05/16 03:59

题意：
一共给你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);        }    }}

0 0