P3275 [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个小朋友分到的糖果；

 

输出格式：

 

输出一行，表示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

#include <bits/stdc++.h>
using namespace std;
#define LL long long
int n,k;
struct haha{
	int v,w,next;
}we[300010];
int head[300010];
int vis[100010];
LL ans[100010];
int times[100010];
int tot;
void add(int u,int v,int w){ 
	++tot;
	we[tot].v=v;
	we[tot].w=w;
	we[tot].next=head[u];
	head[u]=tot;
}
inline int sc(){
    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;
}
bool SPFA(){
	memset(vis,0,sizeof(vis));
	memset(times,0,sizeof(times));
	memset(ans,0,sizeof(ans));
	queue<int>Q;
	vis[0]=1;
	Q.push(0);
	while(!Q.empty()){
		int u=Q.front();
		Q.pop();
		vis[u]=0;
		for(int i=head[u];i!=-1;i=we[i].next){
			int v=we[i].v,w=we[i].w;
			if(ans[v]<ans[u]+w){
				ans[v]=ans[u]+w;
				if(vis[v]==0){
					times[v]++;
					if(times[v]==n){
						return 0;
					}
					Q.push(v);
					vis[v]=1;
				}
			}
		}
	}
	return 1;
}
int main(){
	memset(head,-1,sizeof(head));
	n=sc(),k=sc();
	int f=0;
	tot=0;
	int a,u,v;
	for(int i=1;i<=k;i++){
		a=sc(),u=sc(),v=sc();
		if(f==1)
		    continue;
		if(a==1){
			add(u,v,0);
			add(v,u,0);
		}
		else if(a==2){
			add(u,v,1);
			if(u==v){
			    f=1;
			}
		}
		else if(a==3){
			add(v,u,0);
		}
		else if(a==4){
			add(v,u,1);
			if(u==v)
			    f=1;
		}
		else{
			add(u,v,0);
		}
	}
	if(f==1){
	    printf("-1\n");
	    return 0;
	}
	for(int i=n;i>=1;i--){
		add(0,i,1);
	}
	if(SPFA()){
		long long sum=0;
		for(int i=1;i<=n;i++){
			sum+=ans[i];
		}
		printf("%lld\n",sum);
	}
	else{
		printf("-1\n");
	}
}