【bzoj 2330】糖果（差分约束）

传送门biu~
用边表示不等关系，SPFA求最长路。（玄学判环）

#include<bits/stdc++.h>
#define N 100000
using namespace std;
#define int long long
int head[N+5],nex[3*N+5],to[3*N+5],val[3*N+5],tp=0;
int n,k,ans,sum,num[N+5],dis[N+5];
bool b[N+5];
queue<int>q;
inline void add(int x,int y,int v)
{
    nex[++tp]=head[x];
    head[x]=tp;
    to[tp]=y;
    val[tp]=v;
}
inline void init(){
    memset(head,-1,sizeof(head));
    scanf("%lld%lld",&n,&k);
    for(int i=1;i<=k;++i){
        int opt,x,y;
        scanf("%lld%lld%lld",&opt,&x,&y);
        switch(opt){
            case 1 :{
                add(x,y,0);
                add(y,x,0);
                break;
            }
            case 2 :{
                add(x,y,1);
                break;
            }
            case 3 :{
                add(y,x,0);
                break;
            }
            case 4 :{
                add(y,x,1);
                break;
            }
            case 5 :{
                add(x,y,0);
                break;
            }
        }
    }
    for(int i=n;i>=1;--i)    add(0,i,1);
}
inline void solve()
{
    q.push(0);
    b[0]=1;
    while(!q.empty()){
        ++sum;
        if(sum==1000000){
            printf("-1");
            return;
        }
        int x=q.front();q.pop();
        b[x]=0;
        for(int i=head[x];~i;i=nex[i]){
            if(dis[to[i]]<dis[x]+val[i]){
                dis[to[i]]=dis[x]+val[i];
                if(!b[to[i]]){
                    q.push(to[i]);
                    b[to[i]]=1;
                    ++num[to[i]];
                    if(num[to[i]]>n){
                        printf("-1");
                        return;
                    }
                }
            }
        }
    }
    for(int i=1;i<=n;++i)        ans+=dis[i];
    printf("%lld",ans);
}
#undef int
int main(){
    init();
    solve();
    return 0;
}