正文
C++之路进阶——codevs2404（糖果）
小程序：扫一扫查出行【扫一扫了解最新限行尾号】
复制小程序
2404 糖果
2011年省队选拔赛四川
时间限制: 1 s
空间限制: 128000 KB
题目等级 : 大师 Master
题目描述 Description
幼儿园里有N个小朋友，lxhgww老师现在想要给这些小朋友们分配糖果，要求每个小朋友都要分到糖果。但是小朋友们也有嫉妒心，总是会提出一些要求，比如小明不希望小红分到的糖果比他的多，于是在分配糖果的时候，lxhgww需要满足小朋友们的K个要求。幼儿园的糖果总是有限的，lxhgww想知道他至少需要准备多少个糖果，才能使得每个小朋友都能够分到糖果，并且满足小朋友们所有的要求。
输入描述 Input Description
输入的第一行是两个整数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个小朋友分到的糖果；
输出描述 Output Description
输出一行，表示lxhgww老师至少需要准备的糖果数，如果不能满足小朋友们的所有要求，就输出-1
样例输入 Sample Input
5 7
1 1 2
2 3 2
4 4 1
3 4 5
5 4 5
2 3 5
4 5 1
样例输出 Sample Output
11
数据范围及提示 Data Size & Hint
对于30%的数据，保证 N<=100
对于100%的数据，保证 N<=100000
对于所有的数据，保证 K<=100000，1<=X<=5，1<=A, B<=N
代码：
#include<cstdio>
#define maxn 1000100
#include<iostream> using namespace std; int cnt;
long long ans; struct cf
{
int n,m,head[maxn],v[maxn],to[maxn],next[maxn],rvs[maxn],ins[maxn],dis[maxn],q[maxn*];
void insert(int u,int vv,int w) {to[++cnt]=vv;next[cnt]=head[u];head[u]=cnt; v[cnt]=w;}
bool spfa()
{
int t=,w=;
q[]=;
ins[]=;
rvs[]=;
while (t<w)
{
int now=q[t++];
for (int i=head[now];i;i=next[i])
if (dis[to[i]]<dis[now]+v[i])
{
if (++rvs[to[i]]>=n) return ;
dis[to[i]]=dis[now]+v[i];
if (!ins[to [i]]){ins[to[i]]=;q[w++]=to[i];}
}
ins[now]=;
}
return ;
}
int start()
{
scanf("%d%d",&n,&m);
for (int i=;i<=m;i++)
{
int f,a,b;
scanf("%d%d%d",&f,&a,&b);
switch(f)
{
case :insert(a,b,); insert(b,a,); break;
case :if (a==b) {printf("-1\n");return ;} insert(a,b,);break;
case :insert(b,a,);break;
case :if (a==b) {printf("-1\n");return ;} insert(b,a,);break;
case :insert(a,b,); break;
}
}
for (int i=n;i>=;i--) insert(,i,);
if (!spfa()) {printf("-1\n");return ;}
for (int i=;i<=n;i++) ans+=dis[i];
printf("%lld\n",ans);
}
}cf;
int main()
{
cf.start();
return ;
}
