某公司有工作人员x1,x2,…,xn ,他们去做工作y1,y2,…,ym ,每个人都能做其中的几项工作,并且对每一项工作都有一个固定的效率。问能否找到一种合适的工作分配方案,使得总的效率最高。要求一个人只能参与一项工作,同时一项工作也必须由一个人独立完成。不要求所有的人都有工作。
这里其实是要保证每个工作都有人做的,不然就不能跑mcmf了。而这题的图灰常稠密,所以mcmf我去学了个原始对偶。
#pragma GCC optimize(3,"inline","Ofast")
#include<bits/stdc++.h>
using namespace std;
const int N=1010,M=250010;
const int s=0,t=1005;
void read(int &x)
{
char c=getchar();x=0;
while(!isdigit(c))c=getchar();
while(isdigit(c))x=(x<<3)+(x<<1)+c-48,c=getchar();
}
int n,m,k,hd[N],nxt[M*3],to[M*3],cost[M*3],las[M*3],tot=-1,mp[510][510];
int dis[N],q[M],D=0,ans=0;
bool vis[N],inq[N];
void add(int u,int v,int w,int c)
{
nxt[++tot]=hd[u],to[tot]=v,las[tot]=w,cost[tot]=c,hd[u]=tot;
nxt[++tot]=hd[v],to[tot]=u,las[tot]=0,cost[tot]=-c,hd[v]=tot;
}
bool spfa()
{
int nw,hed=N,tail=N+1;
memset(dis,127,sizeof dis);
memset(inq,0,sizeof inq);
dis[s]=0,q[++hed]=s,inq[s]=1;
while(hed>=tail)
{
nw=q[tail++];
for(int i=hd[nw];i!=-1;i=nxt[i])
{
if(las[i]&&dis[to[i]]>dis[nw]+cost[i])
{
dis[to[i]]=dis[nw]+cost[i];
if(!inq[to[i]])
{
if(dis[to[i]]<dis[q[tail]])q[--tail]=to[i];
else q[++hed]=to[i];
}
inq[to[i]]=1;
}
}
inq[nw]=0;
}
for(int i=0;i<=t;i++)
for(int j=hd[i];j!=-1;j=nxt[j])
cost[j]-=dis[to[j]]-dis[i];
D+=dis[t];
return dis[t]<=2e9;
}
int dfs(int pos,int flow)
{
if(pos==t)return ans+=flow*D,flow;
vis[pos]=1;
int l=flow,tp;
for(int i=hd[pos];i!=-1;i=nxt[i])
{
if(las[i]&&!cost[i]&&!vis[to[i]])
{
tp=dfs(to[i],min(l,las[i]));
las[i]-=tp,las[i^1]+=tp;
l-=tp;
if(!l)return flow;
}
}
return flow-l;
}
void mcmf()
{
while(spfa())
{
do{
memset(vis,0,sizeof vis);
}while(dfs(s,2e9));
}
printf("%d",-ans);
}
int main()
{
int u,v,w;
memset(hd,-1,sizeof hd);
memset(nxt,-1,sizeof nxt);
memset(mp,127,sizeof 127);
read(n),read(m),read(k);
for(int i=1;i<=k;i++)
{
read(u),read(v),read(w);
mp[v][u]=min(mp[v][u],-w);
}
for(int i=1;i<=m;i++)
add(s,i,1,0);
for(int i=1;i<=n;i++)
add(i+m,t,1,0);
for(int i=1;i<=m;i++)
for(int j=1;j<=n;j++)
if(mp[i][j]<=2e9)add(i,j+m,1,mp[i][j]);
mcmf();
}