[POJ2135]Farm Tour 做题笔记

题目链接：http://poj.org/problem?id=2135
这是一类比较经典的费用流水题。
1、建图，每条边容量为1，费用为题目所述距离。
2、新建超级源S，S到1连一条容量为2费用为0的边。
3、跑一遍费用流。
事实上如果用的是一次找一条增广路的费用流，还有一个小优化，那就是网络流为2时直接输出cost退出即可。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <queue>
using namespace std;
const int N=2010,M=10100;
int head[N],ver[M<<2],next[M<<2],e[M<<2],c[M<<2];
int tot=1,s,t,n,m;
int d[N],pre[N],pos[N];
bool vis[N];

void add(int u,int v,int w,int cost) {
    ver[++tot]=v;e[tot]=w;c[tot]=cost;next[tot]=head[u];head[u]=tot;
    ver[++tot]=u;e[tot]=0;c[tot]=-cost;next[tot]=head[v];head[v]=tot;
}

bool spfa () {
    queue<int> q;
    memset(vis,0,sizeof(vis));
    memset(pre,-1,sizeof(pre));
    memset(pos,0,sizeof(pos));
    for (int i=0;i<=n+9;i++) d[i]=0x3fffffff; //注意这里，一开始我把n+9错写成了N+9，结果WA了
    q.push(s); d[s]=0; vis[s]=1; pre[s]=s;
    while (!q.empty()) {
        int x=q.front(); q.pop();
        for (int i=head[x];i;i=next[i]) 
            if (e[i]&&d[x]+c[i]<d[ver[i]]) {
                d[ver[i]]=d[x]+c[i];
                pre[ver[i]]=x,pos[ver[i]]=i;
                if (!vis[ver[i]]) q.push(ver[i]),vis[ver[i]]=1;
            }
        vis[x]=0;
    }
    return pre[t]!=-1;
}

int mcf () {
    int cost=0,maxflow=0,aug;
    while (spfa()) {
        aug=0x3fffffff;
        for (int i=t;i!=s;i=pre[i])  
            aug=min(aug,e[pos[i]]);
        maxflow+=aug,cost+=aug*d[t];
        for (int i=t;i!=s;i=pre[i]) 
            e[pos[i]]-=aug,e[pos[i]^1]+=aug;
    }
    return cost;
}

int main () {
    int u,v,w;
    scanf("%d%d",&n,&m);
    for (int i=1;i<=m;i++) {
        scanf("%d%d%d",&u,&v,&w);
        add(u,v,1,w);add(v,u,1,w);
    }
    s=0,t=n;
    add(s,1,2,0);
    printf("%d",mcf());
}