hdu4411(费用流)

链接:点击打开链接

题意:给n+1个点,m条边的无向图。起点为0,k个人初始在起点,去遍历图使得每个点至少被一人走过并且遍历i点时i-1必须已经被遍历,最后k人要回到起点。输出k个人最小的路径和

代码:

#include <math.h>
#include <vector>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
const int INF=0x3f3f3f3f;
struct node{
    int to,cap,cost,rev;
};
int S,T,V;
int g[505][505],dis[505],prevv[505],preve[505];
vector<node> G[505];
void addedge(int from,int to,int cap,int cost){
    G[from].push_back((node){to,cap,cost,G[to].size()});
    G[to].push_back((node){from,0,-cost,G[from].size()-1});
}
int min_cost_flow(int s,int t,int f){
    int i,d,v,ans,sign;
    ans=0;
    while(f>0){
        fill(dis,dis+V,INF);
        dis[s]=0;
        sign=1;
        while(sign){
        sign=0;
        for(v=0;v<V;v++){
            if(dis[v]==INF)
            continue;
            for(i=0;i<G[v].size();i++){
                node &e=G[v][i];
                if(e.cap>0&&dis[e.to]>dis[v]+e.cost){
                    dis[e.to]=dis[v]+e.cost;
                    prevv[e.to]=v;
                    preve[e.to]=i;
                    sign=1;
                }
            }
        }
        }
        if(dis[t]==INF)
        return -1;
        d=f;
        for(v=t;v!=s;v=prevv[v])
        d=min(d,G[prevv[v]][preve[v]].cap);
        f-=d;
        ans+=d*dis[t];
        for(v=t;v!=s;v=prevv[v]){
            node &e=G[prevv[v]][preve[v]];
            e.cap-=d;
            G[v][e.rev].cap+=d;
        }
    }
    return ans;
}                                               //费用流模板
int main(){
    int n,m,k,i,j,p,u,v,w;
    while(scanf("%d%d%d",&n,&m,&k)!=EOF&&(n||m)){
        V=2*n+4;
        S=2*n+2,T=2*n+3;
        for(i=0;i<=V;i++)
        G[i].clear();
        memset(g,INF,sizeof(g));
        for(i=0;i<m;i++){
            scanf("%d%d%d",&u,&v,&w);
            g[u][v]=min(g[u][v],w);
            g[v][u]=g[u][v];
        }
        for(p=0;p<=n;p++)
        for(i=0;i<=n;i++)
        for(j=0;j<=n;j++)
        g[i][j]=min(g[i][j],g[i][p]+g[p][j]);   //先求出两两间的最短路
        addedge(S,0,k,0);                       //将源点与0相连,容量是k,费用是0
        addedge(0,T,k,0);                       //将汇点与0相连代表不一定适用全部警察
        for(i=1;i<=n;i++){
            for(j=i+1;j<=n;j++)
            addedge(i+n,j,1,g[i][j]);           //将i与j相连(j>i),从而保证了按顺序抓捕
            addedge(0,i,1,g[0][i]);             //因为范围是0~N并且全部能够抓捕,并且是
            addedge(i+n,T,1,g[i][0]);           //按照递增顺序,则一定是按0~N的顺序
            addedge(i,i+n,1,-1000000);          //因为费用流沿最短路增广因此保证了所有点
        }                                       //都会走到,也就是为什么拆点
        printf("%d\n",min_cost_flow(S,T,k)+n*1000000);
    }                                           //将拆点的费用加回来
    return 0;
}


 

你可能感兴趣的:(hdu4411(费用流))