[BZOJ 1202][HNOI2005]狡猾的商人（并查集+前缀和）

最新推荐文章于 2021-08-11 11:43:06 发布

原创最新推荐文章于 2021-08-11 11:43:06 发布 · 247 阅读

0 ·

CC 4.0 BY-SA版权

并查集专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一种利用并查集与前缀和思想来判断账本真实性的算法。通过对比账本中不同时间段的总收入，可以有效检测账本是否被篡改。

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

Description

刁姹接到一个任务，为税务部门调查一位商人的账本，看看账本是不是伪造的。账本上记录了n个月以来的收入情况，其中第i 个月的收入额为Ai(i=1,2,3…n-1,n)，。当 Ai大于0时表示这个月盈利Ai 元，当 Ai小于0时表示这个月亏损Ai 元。所谓一段时间内的总收入，就是这段时间内每个月的收入额的总和。刁姹的任务是秘密进行的，为了调查商人的账本，她只好跑到商人那里打工。她趁商人不在时去偷看账本，可是她无法将账本偷出来，每次偷看账本时她都只能看某段时间内账本上记录的收入情况，并且她只能记住这段时间内的总收入。现在，刁姹总共偷看了m次账本，当然也就记住了m段时间内的总收入，你的任务是根据记住的这些信息来判断账本是不是假的。

Input

第一行为一个正整数w，其中w < 100，表示有w组数据，即w个账本，需要你判断。每组数据的第一行为两个正整数n和m，其中n < 100，m < 1000，分别表示对应的账本记录了多少个月的收入情况以及偷看了多少次账本。接下来的m行表示刁姹偷看m次账本后记住的m条信息，每条信息占一行，有三个整数s，t和v，表示从第s个月到第t个月（包含第t个月）的总收入为v，这里假设s总是小于等于t。

Output

包含w行,每行是true或false,其中第i行为true当且仅当第i组数据，即第i个账本不是假的；第i行为false当且仅当第i组数据，即第i个账本是假的。

Sample Input

2
3 3
1 2 10
1 3 -5
3 3 -15
5 3
1 5 100
3 5 50
1 2 51

Sample Output

true
false

Solution

运用并查集与前缀和思想，Si数组记录点father[i]到点[i]的Ai之和
需要理解的重点在于s[fy]=z-s[y]+s[x]，此处画了个图

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
using namespace std;
int s[1005],father[1005];
int w,n,m;
int find(int x)
{
    if(father[x]==x)return x;
    int f=find(father[x]);
    s[x]+=s[father[x]];
    father[x]=f;
    return f;
}
int main()
{
    scanf("%d",&w);
    while(w--)
    {
        bool flag=true;
        scanf("%d%d",&n,&m);
        memset(s,0,sizeof(s));
        for(int i=0;i<=n;i++)//记得要从0开始，因为这个WA了一次 
        father[i]=i;
        int x,y,z;
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d%d",&x,&y,&z);
            x--;
            int fx=find(x),fy=find(y);
            if(fx!=fy)
            {
                father[fy]=fx;
                s[fy]=z-s[y]+s[x];
            }
            else if(s[y]-s[x]!=z)
            {
                flag=false;
                break;
            }
        }
        if(flag)printf("true\n");
        else printf("false\n");
    }
    return 0;
}