幼儿园里有N个小朋友,lxhgww老师现在想要给这些小朋友们分配糖果,要求每个小朋友都要分到糖果。但是小朋友们也有嫉妒心,总是会提出一些要求,比如小明不希望小红分到的糖果比他的多,于是在分配糖果的时候,lxhgww需要满足小朋友们的K个要求。幼儿园的糖果总是有限的,lxhgww想知道他至少需要准备多少个糖果,才能使得每个小朋友都能够分到糖果,并且满足小朋友们所有的要求。
标签:
幼儿园里有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。
【数据范围】
对于30%的数据,保证 N<=100
对于100%的数据,保证 N<=100000
对于所有的数据,保证 K<=100000,1<=X<=5,1<=A, B<=N
差分约束系统,注意大于、小于、不大于、不小于的区别,等于双向大于等于就可以了。
有一个坑点:0到i加边的时候,要从后往前加,否则会TLE。
#include<iostream> #include<cstdio> #include<cstdlib> #include<cmath> #include<cstring> #include<algorithm> #include<queue> #define F(i,j,n) for(int i=j;i<=n;i++) #define D(i,j,n) for(int i=j;i>=n;i--) #define ll long long #define maxn 100005 using namespace std; int n,m,cnt,head[maxn],vst[maxn],dis[maxn]; ll ans; bool inq[maxn]; struct edge_type{int next,to,v;}e[maxn*3]; queue<int> q; inline int read() { int x=0,f=1;char ch=getchar(); while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();} while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();} return x*f; } inline void add_edge(int x,int y,int v) { e[++cnt]=(edge_type){head[x],y,v}; head[x]=cnt; } inline bool spfa() { q.push(0);inq[0]=true;vst[0]=1; while (!q.empty()) { int x=q.front();q.pop(); for(int i=head[x];i;i=e[i].next) { int y=e[i].to; if (dis[x]+e[i].v>dis[y]) { dis[y]=dis[x]+e[i].v; if (++vst[y]>=n) return false; if (!inq[y]){inq[y]=true;q.push(y);} } } inq[x]=false; } return true; } int main() { n=read();m=read(); F(i,1,m) { int flag=read(),x=read(),y=read(); if (flag==1) add_edge(x,y,0),add_edge(y,x,0); else if (flag==2) { if (x==y){puts("-1");return 0;} add_edge(x,y,1); } else if (flag==3) add_edge(y,x,0); else if (flag==4) { if (x==y){puts("-1");return 0;} add_edge(y,x,1); } else add_edge(x,y,0); } D(i,n,1) add_edge(0,i,1);//这里写F(i,1,n)就会TLE...坑的不得了... if (!spfa()){puts("-1");return 0;} F(i,1,n) ans+=dis[i]; printf("%lld\n",ans); return 0; }
标签:
原文地址:http://blog.csdn.net/aarongzk/article/details/51108072