洛谷P3257糖果

题目思路
差分约束。由于求的是最小值,所以我们需要求最长路径。题目中有五种情况

  • X = 1,A和B的糖果一样多表示关系为(A>=B,B>=A)
  • X = 2,A的糖果必须比B少,表示为(B>=A+1)
  • X = 3,A的糖果不少于B,表示为(A>=B)
  • X = 4,A的糖果必须多余B,表示为(A >= B +1)
  • X = 5,A的糖果不多余B,表示为A<=B

由于每个小朋友都要分到糖果,所以对于每个X都要大于等于1,所以我们建立一个超级源点(X_0),每个点的取值为(X >= X_0 + 1)
注意
题目的数据范围最大的时候,需要用long long,无解的情况就是存在负环,数据量太大spfa会超时,可以把队列换成栈
实现代码

#include <iostream>
#include <cstring>
#include <cstdio>

using namespace std;

const int N = 100010, M = 300010;

typedef long long LL;

int h[N], e[M], ne[M], w[M], idx;
int n, m;
LL dist[N];
bool st[N];
int q[N];
int cnt[N];


void add(int a,int b,int c)
{
    e[idx] = b, ne[idx] = h[a], w[idx] = c, h[a] = idx ++;
}

bool spfa()
{
    memset(dist,-0x3f,sizeof dist);
    dist[0] = 0;
    st[0] = true;
    int hh = 0, tt = 1;
    q[0] = 0;
    while(hh != tt)
    {
        int t = q[-- tt];
        st[t] = false;

        for(int i = h[t];~i; i = ne[i])
        {
            int j = e[i];
            if(dist[j] < dist[t] + w[i])
            {
                dist[j] = dist[t] + w[i];
                cnt[j] = cnt[t] + 1;
                if(cnt[j] >= n + 1) return false;
                if(!st[j])
                {
                    q[ tt ++] = j;
                    st[j] = true;
                }
            }
        }
    }
    return true;
}


int main()
{
    scanf("%d%d",&n, &m);
    memset(h,-1,sizeof h);
    for(int i = 0;i < m;i ++)
    {
        int x, a, b;
        scanf("%d%d%d",&x,&a,&b);
        if(x == 1) add(a,b,0),add(b,a,0);
        else if(x == 2) add(a,b,1);
        else if(x == 3) add(b,a,0);
        else if(x == 4) add(b,a,1);
        else add(a,b,0);
    }

    for(int i = 1;i <= n;i++) add(0,i,1);

    long long res = 0;

    if(!spfa()) puts("-1");
    else
    {
        for(int i = 1;i <= n;i++) res += dist[i];
        printf("%lld", res);
    }
    return 0;
}
原文地址:https://www.cnblogs.com/zykBlog/p/13860451.html