BZOJ 1202 [HNOI2005]狡猾的商人：並查集（維護距離）

題目連結：http://www.lydsy.com/JudgeOnline/problem.php?id=1202

題意：

　　有一個賬本，記錄了n個月的盈虧。

　　每個月的數值：正為盈，負為虧。

　　你知道m個這個賬本的區間和[x[i],y[i]]。

　　問你這個賬本是真是假。

 

題解：

　　如果已知區間和[a,b],[b,c]，那麼就可以算出區間和[a,c]。

　　而唯一判斷賬本真假的方法，就是看有沒有某個給定的區間和A[a,c]與推出來的區間和B[a,c]不相等。如果不相等，賬本為假。

 

　　我們常用字首和來處理區間和。

　　但此題僅給出點與點之間的差值，因此要用並查集維護：

　　　　某一連通塊內各節點到根節點的差值dis[i]。

　　　　如果兩點a,b在同一連通塊內，則區間和為dis[b] - dis[a-1]。

　　根節點dis = 0.

　　每次要用到某個dis[x]時，要在之前執行一遍find(x)，以更新x的真正父親，也順便將dis[x]改為了它與真父親的差值。

　　讓px認py作爹時，要更新dis[px] = px到py的差值 = dis[y]-w-dis[x].

 

　　特別要注意字首和方向：

　　　　“從根到節點的字首和方向”與“認爹箭頭方向”相反。

 

AC Code:

#include <iostream>
#include <stdio.h>
#include <string.h>
#define MAX_N 105

using namespace std;

int n,m,t;
int dis[MAX_N];
int par[MAX_N];
bool flag;

void init_union_find()
{
    memset(dis,0,sizeof(dis));
    for(int i=0;i<=n;i++)
    {
        par[i]=i;
    }
}

int find(int x)
{
    if(par[x]==x) return x;
    int t=find(par[x]);
    dis[x]+=dis[par[x]];
    return par[x]=t;
}

void unite(int x,int y,int w)
{
    int px=find(x);
    int py=find(y);
    if(px==py) return;
    dis[px]=dis[y]-w-dis[x];
    par[px]=py;
}

bool same(int x,int y)
{
    return find(x)==find(y);
}

void work()
{
    cin>>n>>m;
    int x,y,w;
    flag=true;
    init_union_find();
    for(int i=0;i<m;i++)
    {
        cin>>x>>y>>w;
        if(same(x-1,y))
        {
            if(dis[y]-dis[x-1]!=w)
            {
                flag=false;
                return;
            }
        }
        else unite(x-1,y,w);
    }
}

void print()
{
    if(flag) cout<<"true"<<endl;
    else cout<<"false"<<endl;
}

int main()
{
    cin>>t;
    while(t--)
    {
        work();
        print();
    }
}