LA6266 Admiral 费用流

直接最大流。。。。由于只用找两条路径，spfa跑两次就可以了。

#include <cstdio>
#include <queue>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
#define MAXN 310000
#define MAXM 310000
#define INF 0x3f3f3f3f
struct node
{
    int u,v,f,c,next;
}e[MAXM],tree[MAXM];
int k,head[MAXN],pre[MAXN],dist[MAXN],vis[MAXN];
int next[MAXN],first[MAXN],to[MAXN],cost[MAXN],flow[MAXN],fa[MAXN];
int en,s,t,maxflow,mincost,m,n,num; //s源点，t汇点
void add(int u,int v,int c,int f)
{
    e[en].u=u;
    e[en].v=v;
    e[en].c=c;
    e[en].f=f;
    e[en].next=head[u];
    head[u]=en++;
    e[en].u=v;
    e[en].v=u;
    e[en].c=-c;
    e[en].f=0;
    e[en].next=head[v];
    head[v]=en++;
}
int spfa()
{
    int i,u,v;
    for(i=0;i<=t;i++)
        pre[i]=-1,vis[i]=0,dist[i]=INF;
    dist[s]=0;
    vis[s]=1;
    queue<int>q;
    q.push(s);
    while(!q.empty())
    {
        u=q.front();
        q.pop();
        for(i=head[u];i!=-1;i=e[i].next)
        {
            v=e[i].v;
            if(e[i].f>0&&dist[u]+e[i].c<dist[v])
            {
                dist[v]=dist[u]+e[i].c;
                pre[v]=i;
                if(!vis[v])
                {
                    vis[v]=1;
                    q.push(v);
                }
            }
        }
        vis[u]=0;
    }
    if(dist[t]==INF)
        return 0;
    return 1;
}
void add2()
{
    int v;
    int maxf=INF;
    for(v=pre[t];~v;v=pre[e[v].u])
        maxf=min(maxf,e[v].f);
    for(v=pre[t];~v;v=pre[e[v].u])
    {
        e[v].f-=maxf;
        e[v^1].f+=maxf;

    }
    maxflow+=maxf;
    mincost+=maxf*dist[t];
}
void init()
{
    maxflow=0;
    mincost=0;
    s=1;
    t=n*2;
    memset(head,-1,sizeof(head));
}

int main()
{
    int i,j,a,b,c,d;
    while(~scanf("%d%d",&n,&m))
    {
        init();
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d%d",&a,&b,&c);
            add(a+n,b,c,1);
        }
        for(int i=2;i<n;i++)
        {
            add(i,i+n,0,1);
        }
        add(1,1+n,0,10);
        add(n,n+n,0,10);
        for(int i=1;i<=2;i++)//只用找2条路径，所以只用2次
        {
            if(spfa())
                add2();
        }
        printf("%d\n",mincost);
    }

    return 0;
}
/*
3 10 2 3 3 2
5
6
7
*/