刷题集-分配工作

某公司有工作人员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();
}