bzoj 3373: [Usaco2004 Mar]Lying Livestock 说谎的牲畜

#include<stdio.h>
#include<iostream>
using namespace std;
const int N=105;
const int M=1005;
int n,m,i,j,ans,w,a[M],x[M],y[M],dis[N],s[N],p[N],g[M];
int tot,head[N],Next[M],to[M],v[M];
void add(int x,int y,int z)
{
    tot++;
    to[tot]=y;
    v[tot]=z;
    Next[tot]=head[x];
    head[x]=tot;
}
int spfa(int X)
{
    int x,i;
    w=0;
    for(i=1;i<=n;i++)
        dis[i]=0,p[i]=1,s[i]=0,g[++w]=i;
    while(w)
    {
        x=g[w--];
        if(s[x]>=n) return 0;
        p[x]=0;
        for(i=head[x];i!=-1;i=Next[i])
            if(dis[x]+v[i]<dis[to[i]])
        {
            dis[to[i]]=dis[x]+v[i];
            if(p[to[i]]==0)
            {
                p[to[i]]=1;
                g[++w]=to[i];
                s[to[i]]++;
            }
        }
    }
    return 1;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(i=1;i<=m;i++)
        scanf("%d%d%d",&a[i],&x[i],&y[i]);
    for(i=1;i<=n;i++)
    {
        tot=0;
        for(j=1;j<=n;j++) head[j]=-1;
        for(j=1;j<=m;j++)
            if(a[j]==i) add(y[j],x[j],0);else add(x[j],y[j],-1);
        ans+=spfa(i);
    }
    cout<<ans;
    return 0;
}