BZOJ1202. [HNOI2005]狡猾的商人（spfa判负环 | 带权并查集）

最新推荐文章于 2019-10-18 22:02:55 发布

原创最新推荐文章于 2019-10-18 22:02:55 发布 · 294 阅读

0 ·

CC 4.0 BY-SA版权

BZOJ 同时被 2 个专栏收录

65 篇文章

订阅专栏

spfa

5 篇文章

订阅专栏

本文介绍了一种用于检测账本是否伪造的算法，通过分析账本中记录的收入情况，利用SPFA算法和带权并查集判断账本数据的一致性和真实性。

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
Hint
Source

思路： 假设x代表前x个月的总收入，那么对于输入的x,y,z，就可以建立(x-1,y,z),(y,x-1,-z)。这样以后，假设得到环，走一个环的权值就代表所有月后的收入，那么跑环得到的权值为正或者为负，都说明是错误的，spfa判正(负)环即可。

#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>

using namespace std;

const int maxn = 2005;
int head[maxn],nex[maxn],val[maxn],to[maxn],cnt[maxn],d[maxn],vis[maxn],tot;
int n,m;

void init()
{
    tot = 0;
    memset(head,0,sizeof(head));
    memset(cnt,0,sizeof(cnt));
}

void add(int x,int y,int z)
{
    to[++tot] = y;
    val[tot] = z;
    nex[tot] = head[x];
    head[x] = tot;
}

bool spfa(int s)
{
    queue<int>q;
    memset(d,0x3f,sizeof(d));
    memset(vis,0,sizeof(vis));
    d[s] = 0;vis[s] = 1;
    q.push(s);
    while(!q.empty())
    {
        int x = q.front();q.pop();
        vis[x] = 0;
        if(++cnt[x] > n)return false;
        for(int i = head[x];i;i = nex[i])
        {
            int v = to[i],z = val[i];
            if(d[v] > d[x] + z)
            {
                d[v] = d[x] + z;
                if(!vis[v]){q.push(v);vis[v] = 1;}
            }
        }
    }
    return true;
}

int main()
{
    int T;scanf("%d",&T);
    while(T--)
    {
        init();
        scanf("%d%d",&n,&m);
        for(int i = 1;i <= m;i++)
        {
            int x,y,z;scanf("%d%d%d",&x,&y,&z);
            add(x - 1,y,z);add(y,x - 1,-z);
        }
        int flag = 1;
        for(int i = 0;i <= n;i++)
        {
            if(!cnt[i])
            {
                if(!spfa(i))
                {
                    printf("false\n");
                    flag = 0;
                    break;
                }
            }
        }
        if(flag)printf("true\n");
    }
    return 0;
}

带权并查集写法

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

int v[1005],fa[1005];

int findset(int x)
{
    if(fa[x] == x)return x;
    int t = findset(fa[x]);
    v[x] += v[fa[x]];
    fa[x] = t;
    return fa[x];
}

int Union(int x,int y,int z)
{
    int rx = findset(x),ry = findset(y);
    if(rx != ry)
    {
        fa[rx] = ry;
        v[rx] = v[y] - v[x] - z; //这一步可以用类似向量偏移的思想，画图看出来
    }
    else if(v[y] - v[x] != z)return false;
    return true;
}

int main()
{
    int T;scanf("%d",&T);
    while(T--)
    {
        int n,m;scanf("%d%d",&n,&m);
        for(int i = 0;i <= n;i++)
        {
            v[i] = 0;
            fa[i] = i;
        }
        int flag = 0;
        for(int i = 1; i <= m;i++)
        {
            int x,y,z;scanf("%d%d%d",&x,&y,&z);
            if(Union(x - 1,y,z))continue;
            else{
                flag = 1;
                break;
            }
        }
        if(flag)
        {
            printf("false\n");
        }
        else printf("true\n");
    }
    return 0;
}