【SCOI2011】糖果 差分约束

题目描述

　　幼儿园里有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个小朋友分到的糖果；

数据范围

对于30%的数据，保证 N<=100
对于100%的数据，保证 N<=100000
对于所有的数据，保证 K<=100000，1<=X<=5，1<=A, B<=N

样例输入

5 7
1 1 2
2 3 2
4 4 1
3 4 5
5 4 5
2 3 5
4 5 1

样例输出

11

解题思路

差分约束，注意应该用最长路。。。S向每一个点连条权值1的边（每人都有糖），如果连0最后+n似乎为错。。。还有就是如果op=2或4并且x=y就直接输出-1，不然会TLE

代码

#include <bits/stdc++.h>
#define Maxn 400005
using namespace std;
inline long long Getint(){long long x=0,f=1;char ch=getchar();while('0'>ch||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while('0'<=ch&&ch<='9'){x=x*10+ch-'0';ch=getchar();}return x*f;}
long long N,S=0,tot[Maxn],vis[Maxn],dis[Maxn],h[Maxn],cnt=0;
struct node{long long to,next,v;}e[Maxn*3];
void AddEdge(long long x,long long y,long long v){e[++cnt]=(node){y,h[x],v};h[x]=cnt;}
bool SPFA(){
    for(int i=0;i<Maxn;i++)dis[i]=-1LL<<61;
    memset(vis,0,sizeof(vis));
    memset(tot,0,sizeof(tot));
    queue<long long>Q;
    Q.push(S);
    dis[S]=0;
    while(Q.size()){
        long long x=Q.front();
        vis[x]=false;
        Q.pop();
        for(long long p=h[x];p;p=e[p].next){
            long long y=e[p].to;
            if(dis[y]<dis[x]+e[p].v){
                dis[y]=dis[x]+e[p].v;
                tot[y]++;
                if(tot[y]==N)return false;
                if(!vis[y]){
                    vis[y]=true;
                    Q.push(y);
                }
            }
        }
    }
    for(long long i=1;i<N;i++)if(dis[i]<-1LL<<60)return false;
    return true;
}
int main(){
    long long n=Getint(),k=Getint();
    for(long long i=n;i;i--)
        AddEdge(S=0,i,1);
    N=n+1;
    while(k--){
        long long op=Getint(),x=Getint(),y=Getint();
        if(op==1)AddEdge(x,y,0),AddEdge(y,x,0);
        if(op==2)AddEdge(x,y,1);
        if(op==3)AddEdge(y,x,0);
        if(op==4)AddEdge(y,x,1);
        if(op==5)AddEdge(x,y,0);
        if(op==2||op==4)if(x==y){cout<<-1;return 0;}
    }
    if(SPFA()){
        long long Min=1LL<<60,Ans=0;
        for(long long i=1;i<=n;i++)Ans+=dis[i];
        cout<<Ans<<"\n";
    }else cout<<-1;
    return 0;
}

转载于:https://www.cnblogs.com/Cedric341561/p/6811002.html