BZOJ4690 Never Wait for Weights(带权并查集模板题)

最新推荐文章于 2021-09-13 18:30:43 发布

原创最新推荐文章于 2021-09-13 18:30:43 发布 · 246 阅读

0 ·

CC 4.0 BY-SA版权

ACM 专栏收录该内容

101 篇文章

订阅专栏

本文介绍了一种使用并查集解决物品重量比较问题的方法。通过对每一对物品进行权值更新，可以有效地处理询问两个物品重量差的问题。文章详细解释了如何实现并查集的查找和合并操作，并附带完整的代码示例。

题意：

一共有n个物品；输入：？ a b w 表示b比a重w，！ a b 表示输出b比a重多少；

思路：

对每个？ a b w；建立a、b的并查集，维护dis数组。dis[x]表示x比x的父亲fa[x]重多少；

代码：

#include <bits/stdc++.h>

using namespace std;

int fa[100100];
long long dis[100100];
int n,m;

int fi(int x)
{
    if(fa[x]==x)
        return x;
    int root = fi(fa[x]);
    dis[x] += dis[fa[x]];
    fa[x] = root;
    return root;
}

void uni(int x,int y,int d)
{
    int fx = fi(x);
    int fy = fi(y);
    if(fx!=fy)
    {
        fa[fy] = fx;
        dis[fy] = dis[x]-dis[y]+d;
    }
}

int main()
{
    while(~scanf("%d%d",&n,&m)&&(n!=0||m!=0))
    {
        memset(dis,0,sizeof(dis));
        for(int i = 0;i<=n;i++)
            fa[i] = i;
        while(m--)
        {
            char s[5],op;
            int a,b,w;
            scanf("%s",s);
            op = s[0];
            if(op=='!')
            {
                scanf("%d%d%d",&a,&b,&w);
                uni(a,b,w);
            }
            else
            {
                scanf("%d%d",&a,&b);
                if (fi(a)!=fi(b))
                    printf("UNKNOWN\n");
                else
                    printf("%d\n",dis[b]-dis[a]);
            }
        }
    }
    return 0;
}