SCOI 糖果 差分约束/SPFA

SCOI 糖果
题目描述
幼儿园里有N个小朋友，lxhgww老师现在想要给这些小朋友们分配糖果，要求每个小朋友都要分到糖果。但是小朋友们也有嫉妒心，总是会提出一些要求，比如小明不希望小红分到的糖果比他的多，于是在分配糖果的时候，lxhgww需要满足小朋友们的K个要求。幼儿园的糖果总是有限的，lxhgww想知道他至少需要准备多少个糖果，才能使得每个小朋友都能够分到糖果，并且满足小朋友们所有的要求。
输入输出格式
输入格式：

输入的第一行是两个整数N，K。接下来K行，表示这些点需要满足的关系，每行3个数字，X，A，B。如果X=1， 表示第A个小朋友分到的糖果必须和第B个小朋友分到的糖果一样多；如果X=2， 表示第A个小朋友分到的糖果必须少于第B个小朋友分到的糖果；如果X=3， 表示第A个小朋友分到的糖果必须不少于第B个小朋友分到的糖果；如果X=4， 表示第A个小朋友分到的糖果必须多于第B个小朋友分到的糖果；如果X=5， 表示第A个小朋友分到的糖果必须不多于第B个小朋友分到的糖果；
输出格式：

输出一行，表示lxhgww老师至少需要准备的糖果数，如果不能满足小朋友们的所有要求，就输出-1。

Input Sample
5 7
1 1 2
2 3 2
4 4 1
3 4 5
5 4 5
2 3 5
4 5 1
Output Sample
11

省选题里面比较简单适合练手的差分约束题。 切入点在于对于每个命令其实都可以转化成一个不等式。
比如当 X=1时 0<=(A-B)<=0 我们在这里可以定义 dis(A)为小朋友拿的糖果，那么可以有以下转换
对于 X=1 dis(A)+0<=dis(B) && dis(B)+0<=dis(A)
对于 X=2 dis(A)+1<=dis(B)
对于 X=3 dis(B)+0<=dis(A)
对于 X=4 dis(B)+1<=dis(A)
对于 X=5 dis(A)+0<=dis(B)
可以发现其实这题在转换为图以后就是在求图中的最长路，因为我们在这条路径上松弛时会把路径从小变大以符合题目要求的最小条件，也就是至少有多少糖果。并且注意到每个人至少有一个糖果，那么我们定义超级源点 0，并且和每个点都接上一条权为1的边，以保证最长路至少对于每个点为1. 这题的正权环或者正权自环就会导致题目无解，因为你会发现 这其实是在要求 dis(A)>dis(B) 同时 dis(B)>dis(A)，导致这个环会无限跑下去。
这题的小TIPS 是得反着和超级源点连边…因为听说原数据有一条巨长的链卡SPFA，然后对于 X=2或X=4，如果A==B 那么说明会有自环，自己一直比自己大 就会无限加下去。

code

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <queue>
#define MAXN 300005 
using namespace std;
int N, M;
int u[MAXN],v[MAXN],w[MAXN],nex[MAXN],first[MAXN];
int tot;
int vis[MAXN],d[MAXN],cnt[MAXN];
void add(int x,int y,int z)
{
 u[++tot]=x;
 v[tot]=y;
 w[tot]=z;
 nex[tot]=first[x];
 first[x]=tot;
}
int gi(){
    int a=0;char x=getchar();
    while(x<'0'||x>'9')x=getchar();
    while(x>='0'&&x<='9')a=(a<<3)+(a<<1)+x-48,x=getchar();
    return a;
}
inline bool spfa()
{
 queue<int> q;
 q.push(0);
 d[0]=0;
 while(!q.empty())
 {
  int x=q.front();
  q.pop();
  vis[x]=0;
  for(int i=first[x];i;i=nex[i])
  {
   if(d[v[i]]<d[x]+w[i])
   {
    d[v[i]]=d[x]+w[i];
    if(!vis[v[i]])
    {
     vis[v[i]]=1;
     cnt[v[i]]++;
     if(cnt[v[i]]>=N)
     {
      return 0;
     }
     q.push(v[i]);
    }
   }
  }
 }
 return 1;
}
int main()
{
 int i, j, k;
 N=gi(),M=gi();
 for(i=1;i<=M;i++)
 {
  int x, y, z;
  z=gi(),x=gi(),y=gi();
  if(z==1)
  {
   add(x,y,0);
   add(y,x,0);
  }
  if(z==2)
  {
   if(x==y)
   {
    cout<<-1<<endl;
    return 0;
   }
   add(x,y,1);
  }
  if(z==3)
  {
   add(y,x,0);
  }
  if(z==4)
  {
   add(y,x,1);
   if(x==y)
   {
    cout<<-1<<endl;
    return 0;
   }
  }
  if(z==5)
  {
   add(x,y,0);
  }
 }
 for(i=N;i>=1;i--)
 {
  add(0,i,1);
 }
 if(!spfa())
 {
  cout<<-1<<endl;
  return 0;
 }
 long long ans=0;
 for(i=1;i<=N;i++)
 ans+=d[i];
 cout<<ans<<endl;
}```