題目描述
幼兒園裏有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");
}
}