[SCOI2011]糖果

Posted 2022-01-30 akura

题目描述

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

输入格式

输入的第一行是两个整数 N，K。接下来 KK 行，表示这些点需要满足的关系，每行 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。

---

差分约束裸题。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
#define maxn 100001
#define maxm 100001
using namespace std;
int n,m;
long long ans;
inline int read()
    register int x(0),f(1); register char c(getchar());
    while(c<'0'||'9'<c) if(c=='-') f=-1; c=getchar(); 
    while('0'<=c&&c<='9') x=(x<<1)+(x<<3)+(c^48),c=getchar();
    return x*f;


struct edge
    int to,dis,next;
    edge()
    edge(const int &_to,const int &_dis,const int &_next) to=_to,dis=_dis,next=_next; 
e[maxm<<2];
int head[maxn],k;
inline void add(const int &u,const int &v,const int &w) e[k]=edge(v,w,head[u]),head[u]=k++; 

long long dis[maxn];
int cnt[maxn];
bool inq[maxn];
queue<int> q;
inline bool spfa()
    q.push(0),inq[0]=true;
    while(q.size())
        int u=q.front(); inq[u]=false,q.pop();
        for(register int i=head[u];~i;i=e[i].next)
            int v=e[i].to;
            if(dis[v]<dis[u]+e[i].dis)
                dis[v]=dis[u]+e[i].dis,cnt[v]=cnt[u]+1;
                if(cnt[v]==n+1) return false;
                if(!inq[v]) q.push(v),inq[v]=true;
            
        
    
    return true;


int main()
    memset(head,-1,sizeof head);
    n=read(),m=read();
    for(register int i=1;i<=m;i++)
        int op=read(),u=read(),v=read();
        if(op==1) add(u,v,0),add(v,u,0);
        if(op==2)
            if(u==v) puts("-1"); return 0; 
            add(u,v,1);
        
        if(op==3) add(v,u,0);
        if(op==4)
            if(u==v) puts("-1"); return 0; 
            add(v,u,1);
        
        if(op==5) add(u,v,0);
    
    for(register int i=n;i>=1;i--) add(0,i,1);
    if(!spfa()) puts("-1");
    else
        for(register int i=1;i<=n;i++) ans+=dis[i];
        printf("%lld\n",ans);
    
    return 0;