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 ;
   }