BZOJ3436: 小K的农场

题解:

差分约束的入门题,用spfa刷环就好了。
代码如下:

#include
#include
#include
using namespace std;
const int maxn=100005;
int n,m,tot,dis[maxn],lnk[maxn],w[2*maxn],son[2*maxn],nxt[2*maxn];
bool vis[maxn],pd;
inline int read(){
    int x=0; char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while (ch>='0'&&ch<='9') x=x*10+ch-48,ch=getchar();
    return x;
}
void add(int x,int y,int z){son[++tot]=y; w[tot]=z; nxt[tot]=lnk[x]; lnk[x]=tot;}
bool dfs(int x){
    vis[x]=1;
    for (int j=lnk[x];j;j=nxt[j])
    if (dis[x]+w[j]if (vis[son[j]]) return 0;
        if (!dfs(son[j])) return 0;
    }
    vis[x]=0;
    return 1;
}
int main(){
    n=read(); m=read();
    for (int i=1;i<=m;i++){
        int p=read(),x=read(),y=read();
        if (p==1) {int z=read(); add(x,y,z);}
        if (p==2) {int z=read(); add(y,x,-z);}
        if (p==3) {add(x,y,0); add(y,x,0);}
    }
    memset(dis,63,sizeof(dis)); dis[0]=0;
    for (int i=0;i<=n;i++) add(0,i,0);
    if (dfs(0)==0) printf("No\n"); else printf("Yes\n");
    return 0;
}

你可能感兴趣的:(题解,BZOJ题解)