bzoj 3436: 小K的农场 （差分约束）

题目描述

传送门

题解

建出最短路图，用spfa判是否存在负环。
这道题的话深搜版的spfa要快很多。。。

代码

#include
#include
#include
#include
#include
#include
#define N 200003
#define inf 1000000000
using namespace std;
int tot,nxt[N],point[N],v[N],c[N],dis[N],can[N],cnt[N],n,m;
void add(int x,int y,int z)
{
    tot++; nxt[tot]=point[x]; point[x]=tot; v[tot]=y; c[tot]=z;
}
int spfa()
{
    queue<int> p;
    for (int i=1;i<=n;i++) dis[i]=inf;
    p.push(0); cnt[0]=can[0]=1; dis[0]=0;
    while (!p.empty()) {
        int now=p.front(); p.pop();
        for (int i=point[now];i;i=nxt[i]) 
         if (dis[v[i]]>dis[now]+c[i]) {
            dis[v[i]]=dis[now]+c[i];
            if (!can[v[i]]){
                can[v[i]]=1;
                cnt[v[i]]++;
                if (cnt[v[i]]>n) return 0;
                cnt[v[i]]++;
                p.push(v[i]);
             }
         }
        can[now]=0;
    }
    return 1;
}
int main()
{
    freopen("a.in","r",stdin);
    scanf("%d%d",&n,&m);
    for (int i=1;i<=m;i++) {
        int opt,a,b,c; 
        scanf("%d%d%d",&opt,&a,&b);
        if (opt==1) {
            scanf("%d",&c);
            add(a,b,-c);
        }
        if (opt==2) {
            scanf("%d",&c);
            add(b,a,c);//add(a,b,0);
        }
        if (opt==3) add(a,b,0),add(b,a,0);
    }
    for (int i=1;i<=n;i++) add(0,i,0);
    if (spfa()) printf("Yes\n");
    else printf("No\n");
}