bzoj 2330: [SCOI2011]糖果

刚开始想的拓扑排序去点然后balabala搞一搞然后发现不能处理op=1的情况然后发现是spfa跑最长边....

2330: [SCOI2011]糖果

Time Limit: 10 Sec Memory Limit: 128 MB
Submit: 7278 Solved: 2452
[Submit][Status][Discuss]

Description

幼儿园里有N个小朋友，lxhgww老师现在想要给这些小朋友们分配糖果，要求每个小朋友都要分到糖果。但是小朋友们也有嫉妒心，总是会提出一些要求，比如小明不希望小红分到的糖果比他的多，于是在分配糖果的时候，lxhgww需要满足小朋友们的K个要求。幼儿园的糖果总是有限的，lxhgww想知道他至少需要准备多少个糖果，才能使得每个小朋友都能够分到糖果，并且满足小朋友们所有的要求。

Input

输入的第一行是两个整数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

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

#include <bits/stdc++.h>
#define N 100005
#define ll long long
using namespace std;
int n,m;
vector<pair<int,int> >vec[N];
int read(){
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
    while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
    return f*x;
}
ll dis[N];
bool vis[N];int ci[N];
ll spfa(){
    queue<int>que;
    for(int i=0;i<=n;i++) dis[i]=0;
    for(int i=1;i<=n;i++){
        vec[0].push_back(make_pair(i,1));
    }
    que.push(0);vis[0]=1;ci[0]++;
    while(!que.empty()){
        int t=que.front();que.pop();vis[t]=0;
        for(int i=0;i<vec[t].size();i++){
            int t1=vec[t][i].first;int t2=vec[t][i].second;
            if(dis[t1]<dis[t]+t2){
                dis[t1]=dis[t]+t2;
                if(vis[t1]==0){
                    vis[t1]=1;que.push(t1);ci[t1]++;
                    if(ci[t1]>=n) return -1;
                }
            }
        }
    }
    ll sum=0;
    for(int i=1;i<=n;i++) sum+=dis[i];
    return sum;
}
int main(){
    ios::sync_with_stdio(false);
    n=read();m=read();
    int op,_x,_y;
    for(int i=1;i<=m;i++){
        op=read();_x=read();_y=read();
        if(op==1){
            vec[_x].push_back(make_pair(_y,0));
            vec[_y].push_back(make_pair(_x,0));
        }
        else if(op==2) vec[_x].push_back(make_pair(_y,1));
        else if(op==3) vec[_y].push_back(make_pair(_x,0));
        else if(op==4) vec[_y].push_back(make_pair(_x,1));
        else vec[_x].push_back(make_pair(_y,0));
    }
    printf("%lld
",spfa());
    return 0;
}