LeetCode137周赛 Q4. 放三个车的价值之和最大 II（最小费用最大流做法，练习网络流建图好题）

题意：

解法：

最小费用最大流做法, 主要难点在于如何建图:
源点s,汇点t.
1. 对于选3个点的约束:
s向s'点连流量3,费用0的边
t'向点t连流量3,费用0的边

2. 对于车每行每列只能选一次的约束:
s'向行节点R[i]连流量1,费用0的边
列节点C[i]向t'连流量1,费用0的边

3. 对于选择的节点最大值的约束:
R[i]向C[i]连流量1,费用-a[i][j]的边 

最大流调节下的最小费用, 取反后就是答案了.
图示:

Code：

#define ll long long
const int N=3000+5;
const int M=6e5+5;
const ll inf=1e18;
struct Node{
    int from,to,nt;
    ll cap,flow,cost;
}e[M];
int head[N];
ll cnt;
ll d[N];
bool mark[N];
ll pre[N];
int s,t;
int idx;
ll mincost,maxflow;
void add(int a,int b,ll c,ll d){
    cnt++;e[cnt].nt=head[a];head[a]=cnt;e[cnt].from=a;e[cnt].to=b;e[cnt].cap=c;e[cnt].flow=0;e[cnt].cost=d;
    cnt++;e[cnt].nt=head[b];head[b]=cnt;e[cnt].from=b;e[cnt].to=a;e[cnt].cap=0;e[cnt].flow=0;e[cnt].cost=-d;
}
bool spfa(){
    for(int i=0;i<=idx;i++){
        mark[i]=0;
        d[i]=inf;
    }
    queue<int>q;
    q.push(s);
    d[s]=0;
    mark[s]=1;
    while(!q.empty()){
        int x=q.front();
        q.pop();
        mark[x]=0;
        for(int i=head[x];i!=-1;i=e[i].nt){
            int v=e[i].to;
            if(e[i].cap>e[i].flow&&d[v]>d[x]+e[i].cost){
                d[v]=d[x]+e[i].cost;
                pre[v]=i;
                if(!mark[v]){
                    q.push(v);
                    mark[v]=1;
                }
            }
        }
    }
    return d[t]!=inf;
}
void ek(){
    mincost=maxflow=0;
    while(spfa()){
        ll k=inf;
        for(int i=t;i!=s;i=e[pre[i]].from){
            k=min(k,e[pre[i]].cap-e[pre[i]].flow);
        }
        for(int i=t;i!=s;i=e[pre[i]].from){
            e[pre[i]].flow+=k;
            e[pre[i]^1].flow-=k;
        }
        maxflow+=k;
        mincost+=k*d[t];
    }
}
ll R[N];
ll C[N];
class Solution {
public:
    long long maximumValueSum(vector<vector<int>>& a) {
        int n=a.size();
        int m=a[0].size();
        // init node
        idx=0;
        s=++idx;t=++idx;
        int s1=++idx;
        int t1=++idx;
        for(int i=0;i<n;i++){
            R[i]=++idx;
        }
        for(int i=0;i<m;i++){
            C[i]=++idx;
        }
        // clear graph 
        for(int i=0;i<=idx;i++){
            head[i]=-1;
        }
        cnt=-1;
        // init edge
        add(s,s1,3,0);
        add(t1,t,3,0);
        for(int i=0;i<n;i++){
            add(s1,R[i],1,0);
        }
        for(int j=0;j<m;j++){
            add(C[j],t1,1,0);
        }
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                add(R[i],C[j],1,-a[i][j]);
            }
        }
        // MCMF
        ek();
        // output
        long long ans=-mincost;
        return ans;
    }
}

---