bzoj 5154: [Tjoi2014]匹配 费用流+最短路

题意

有N个单身的男孩和N个单身女孩，男孩i和女孩j在一起得到的幸福值为Hij。一个匹配即对这N个男孩女孩的安排：每个男孩恰好有一个女朋友，每个女孩恰好有一个男朋友。一个匹配的幸福值即这N对男女朋友的幸福值的和。经典的问题是计算幸福值最大的匹配，即完美匹配。然而完美匹配有时候并不唯一，你需要计算，对于所有的完美匹配，其交集是什么。
N ≤ 80

分析

先找到某个最大权匹配，然后枚举每条匹配边(x,y)。
若该边不是必须在最大权匹配中的点，则必然满足存在一条从x到y权值为边权的相反数的增广路。
那么每次跑一次最短路来判断即可。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;

const int N=205;
const int inf=1000000000;

int n,cnt,last[N],dis[N],pre[N],s,t,ans;
bool vis[N];
struct edge{int to,next,c,w;}e[N*N];
queue<int> que;

void addedge(int u,int v,int c,int w)
{
    e[++cnt].to=v;e[cnt].c=c;e[cnt].w=w;e[cnt].next=last[u];last[u]=cnt;
    e[++cnt].to=u;e[cnt].c=0;e[cnt].w=-w;e[cnt].next=last[v];last[v]=cnt;
}

bool spfa(int s,int t)
{
    for (int i=0;i<=n*2+1;i++) dis[i]=-inf;
    dis[s]=0;vis[s]=1;que.push(s);
    while (!que.empty())
    {
        int x=que.front();que.pop();
        for (int i=last[x];i;i=e[i].next)
            if (e[i].c&&dis[x]+e[i].w>dis[e[i].to])
            {
                pre[e[i].to]=i;
                dis[e[i].to]=dis[x]+e[i].w;
                if (!vis[e[i].to]) que.push(e[i].to),vis[e[i].to]=1;
            }
        vis[x]=0;
    }
    return dis[t]>-inf;
}

void mcf()
{
    int x=t;ans+=dis[t];
    while (x!=s)
    {
        e[pre[x]].c--;
        e[pre[x]^1].c++;
        x=e[pre[x]^1].to;
    }
}

int main()
{
    scanf("%d",&n);s=0;t=n*2+1;cnt=1;
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
        {
            int x;scanf("%d",&x);
            addedge(i,j+n,1,x);
        }
    for (int i=1;i<=n;i++) addedge(s,i,1,0),addedge(i+n,t,1,0);
    while (spfa(s,t)) mcf();
    printf("%d\n",ans);
    for (int i=2;i<=cnt;i+=2)
    {
        if (e[i].c) continue;
        int x=e[i^1].to,y=e[i].to;
        if (x==s||y==t) continue;
        e[i^1].c=0;
        if (spfa(x,y)&&dis[y]!=e[i].w) printf("%d %d\n",x,y-n);
        e[i^1].c=1;
    }
    return 0;
}