计蒜客-Our Journey of Dalian Ends 拆点+最小费用最大流

题目

Our Journey of Dalian Ends

题解

需要用到最小费用最大流的知识,不会的话可以先学习一下。
有了这个前置知识,这道题就很简单了。

首先由于每个城市只能经过一次,所以要先进行拆点。将一个城市拆成两个点,这两个点单向流动,容量为1,路径长度为0。

点拆好后,要再加两个点,三条路径。一个超级源点,连接上海,流量为2,路径长度为1;一个超级汇点,分别连大连和上海,流量为1,路径长度为0。

最后跑一个从超级源点到超级汇点的最小费用最大流就好了。如果答案流量为2,就把费用输入,但如果流量小于2,就输出-1。

代码

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;

const int MAX = 50000;
struct Edge
{
    int from,to;
    long long cap,flow,cost;
    Edge(int u,int v,long long ca,long long f,long long co):from(u),to(v),cap(ca),flow(f),cost(co){};
};

struct MCMF
{
    int n;
    vector edges;
    vector G[MAX];
    int inq[MAX];//是否在队列中
    long long d[MAX];//距离
    int p[MAX];//上一条弧
    long long a[MAX];//可改进量

    void init(int n)//初始化
    {
        this->n=n;
        for(long long i=0;i<=n;i++)
            G[i].clear();
        edges.clear();
    }

    void addedge(int from,int to,long long cap,long long cost)//加边
    {
        edges.push_back(Edge(from,to,cap,0,cost));
        edges.push_back(Edge(to,from,0,0,-cost));
        int m = edges.size();
        G[from].push_back(m-2);
        G[to].push_back(m-1);
    }

    bool SPFA(int s,int t,long long &flow,long long &cost)//寻找最小费用的增广路,使用引用同时修改原flow,cost
    {
        for(int i=1;i<=n;i++)
            d[i] = LLONG_MAX;
        memset(inq,0,sizeof(inq));
        d[s]=0;inq[s]=1;p[s]=0;a[s]=LLONG_MAX;
        queue Q;
        Q.push(s);
        while(!Q.empty()) {
            int u=Q.front();
            Q.pop();
            inq[u]--;
            for(int i=0;ie.flow && d[e.to]>d[u]+e.cost)//满足可增广且可变短
                {
                    d[e.to]=d[u]+e.cost;
                    p[e.to]=G[u][i];
                    a[e.to]=min(a[u],e.cap-e.flow);
                    if(!inq[e.to])
                    {
                        inq[e.to]++;
                        Q.push(e.to);
                    }
                }
            }
        }
        if(d[t]==LLONG_MAX)
            return false;//汇点不可达则退出
        flow += a[t];
        cost += d[t] * a[t];
        int u = t;
        while(u != s)//更新正向边和反向边
        {
            edges[p[u]].flow += a[t];
            edges[p[u]^1].flow -= a[t];
            u = edges[p[u]].from;
        }
        return true;
    }

    long long MincotMaxflow(int s,int t)
    {
        long long flow=0,cost=0;
        while(SPFA(s,t,flow,cost));
        if(flow < 2)
            return -1;
        return cost;
    }
};

mapm;
int main(){
    int t;
    cin >> t;
    while(t--){
        MCMF mcmf;
        string a,b;
        long long len;
        int n;

        m.clear();
        cin >> n;

        int N = 2 * n + 10;
        int cnt = 6;

        mcmf.init(N * 2);

        m["Shanghai"] = 3;
        mcmf.addedge(3,3 + N,2,0);
        m["Xian"] = 4;
        mcmf.addedge(4,4 + N,1,0);
        m["Dalian"] = 5;
        mcmf.addedge(5,5 + N,1,0);

        for(int i=0;i> a >> b;
            scanf("%lld",&len);
            if(m.find(a) == m.end()){
                m[a] = cnt++;
                mcmf.addedge(m[a],m[a] + N,1,0);
            }
            if(m.find(b) == m.end()){
                m[b] = cnt++;
                mcmf.addedge(m[b],m[b] + N,1,0);
            }
            mcmf.addedge(m[a] + N,m[b],1,len);
            mcmf.addedge(m[b] + N,m[a],1,len);
        }

        mcmf.addedge(1,m["Shanghai"],2,0);
        mcmf.addedge(m["Xian"] + N,2,1,0);
        mcmf.addedge(m["Dalian"] + N,2,1,0);

        cout << mcmf.MincotMaxflow(1,2) << endl;
    }

    return 0;
}

你可能感兴趣的:(ACM/ICPC)