糖果【洛谷P3275】

【题目描述】：

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

PS：出题人/搬题人太毒，建图顺序不对会导致#5会TLE/WA。。。
【输入描述】：

输人的第一行是两个整数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。
【样例输入】：

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

【样例输出】：

11

【时间限制、数据范围及描述】：

时间：1s 空间：128M

对于30%的数据，保证N<=100。

对于100%的数据，保证N<=100,000。

对于所有的数据，保证K<=100,000；1<=X<=5；1<=A,B<=N。

思路
注意要用longlong,还要特判-1的情况

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=1000005;
int n,tt,h[N],hh[N],to[N],cnt,w[N],dis[N],tot[N];
bool vis[N];
queue<int>q;
long long ans;
void add(int u,int v,int c){
	to[++cnt]=v;
	hh[cnt]=h[u];
	h[u]=cnt;
	w[cnt]=c;
}
void spfa()
{
	vis[0]=1;
	q.push(0);
    while(!q.empty()){
        int u=q.front();
		q.pop();
		vis[u]=0;
        if(tot[u]==n-1){
			printf("-1");
			return;
		}
        tot[u]++;
        for(int i=h[u];i;i=hh[i]){
            if(dis[to[i]]<dis[u]+w[i]){
                dis[to[i]]=dis[u]+w[i];
                if(!vis[to[i]]){
					vis[to[i]]=1;
				}	
				q.push(to[i]);
            }
        }    
    }
}
int main()
{
	scanf("%d%d",&n,&tt);
    int u,v,x;
    while(tt--){
        scanf("%d%d%d",&x,&u,&v);
        if(x==1)
		{
			add(u,v,0);
			add(v,u,0);
		}	
        if(x==2){
            if(u==v){
				printf("-1");
				return 0;
			}
            add(u,v,1);
        }
        if(x==3)
			add(v,u,0);	
        if(x==4)
		{	
            if(v==u)
			{
				printf("-1");
				return 0;
			}
            add(v,u,1);
        }
        if(x==5)
			add(u,v,0);	
    }
    for(int i=1;i<=n;i++)
		add(0,i,1);
	spfa();
    for(int i=1;i<=n;i++)
		ans+=dis[i];
    printf("%lld",ans);
    return 0;
}