题意:
解法:
最小费用最大流做法, 主要难点在于如何建图:
源点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;
}
}