P3967 [TJOI2014]匹配 KM算法

这道题首先要实现KM算法求出权值；

然后要求我们计算出所有的完美匹配中的交集，也就是说，必须出现在完美匹配当中的男女朋友关系

所以要求出交集，我们可以采用暴力的方式

　　由于KM算法（我用的是优化后的模板 0（n^3））题目给出的范围也就80，所以多一层暴力也不会超时

　　所以我们把第一次匹配中存在的男女生配偶数组再copy到另一个数组sto【】当中

　　然后将每一组中的权值置为0，再算KM算法，假如总值不变，则证明不在交集当中，变小就证明在。

　　　　变大呢？对不起，不可能变大的

然后最后排序输出就好

#include<bits/stdc++.h>
using namespace std;
//Data
typedef long long ll;
const int N=500;
const int inf=0x3f3f3f3f;
int n,m,e[N+7][N+7];
int sto[N];
//KM
int mb[N+7],vb[N+7],ka[N+7],kb[N+7],p[N+7],c[N+7];
int qf,qb;
struct node
{
    int l,r;
}last[N];
void Bfs(int u){
    int a,v=0,vl=0,d;
    for(int i=1;i<=n;i++) p[i]=0,c[i]=inf;
    mb[v]=u;
    do {
        a=mb[v],d=inf,vb[v]=1;
        for(int b=1;b<=n;b++)if(!vb[b]){
            if(c[b]>ka[a]+kb[b]-e[a][b])
                c[b]=ka[a]+kb[b]-e[a][b],p[b]=v;
            if(c[b]<d) d=c[b],vl=b;
        }
        for(int b=0;b<=n;b++)
            if(vb[b]) ka[mb[b]]-=d,kb[b]+=d;
            else c[b]-=d;
        v=vl;
    } while(mb[v]);
    while(v) mb[v]=mb[p[v]],v=p[v];
}
ll KM()
{
    for(int i=1;i<=n;i++)
        mb[i]=ka[i]=kb[i]=0;
    for(int a=1;a<=n;a++){
        for(int b=1;b<=n;b++)
            vb[b]=0;
        Bfs(a);
    }
    ll res=0;
    for(int b=1;b<=n;b++) res+=e[mb[b]][b];
    return res;
}

//Main
bool cmp(node x,node y)
{
    return x.l<y.l;
}
int main()
{
    scanf("%d",&n);
    for(int a=1;a<=n;a++)
        for(int b=1;b<=n;b++)
            e[a][b]=-inf;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            int tmp;
            scanf("%d",&tmp);
            e[i][j]=tmp;
        }
    }
    ll ans=KM();
    printf("%lld\n",ans);

    //存储的是第i个女生的男朋友编号
    for(int i=1;i<=n;i++) sto[i]=mb[i];

    int cot=0;
    for(int i=1;i<=n;i++){
        int v=i;
        int u=sto[i];
        int tmp=e[u][v];
        e[u][v]=0;
        if(KM()<ans){
            cot++;
            last[cot].l=u;
            last[cot].r=v;
        }
        e[u][v]=tmp;
    }
    sort(last+1,last+1+cot,cmp);
    for(int i=1;i<=cot;i++)
        printf("%d %d\n",last[i].l,last[i].r);
    return 0;
}