HDU 3718 Similarity（最大权匹配-KM）

Description

给出$n$个大写字母$a_1,...,a_n$和$n$个大写字母$b_1,...,b_n$，已知$a_1,...,a_n$中有$k$个不同的大写字母，可以把$b_1...,b_n$中相同的字母同时变成另一个字母，但是如果两个位置的字母之前不同则变化后也要不同，问变化过后满足$a_i=b_i$的$i$的数量除以$n$的最大值

Input

第一行一整数$T$表示用例组数，每组用例首先输入三个整数$n,k,m$，之后输入$n$个大写字母$a_1,...,a_n$，最后$m$行每行输入$n$个大写字母$b_1,...,b_n$表示一组查询，保证$b_1,...,b_n$中至多有$k$个不同的大写字母

$(1\le T\le 15,1\le n\le 10^4,1\le k\le 26,1\le m< 30)$

Output

对于每组查询输出满足$a_i=b_i$的$i$的数量除以$n$的最大值，保留小数点后四位

Sample Input

2
10 3 3
A A B A B B C C C C
F F E F E E D D D D
X X X Y Y Y Y Z Z Z
S T R S T R S T R S
3 2 2
A B A
C D C
F F E

Sample Output

1.0000
0.7000
0.5000
1.0000
0.6667

Solution

问题转化为把$b$中的$x$种字母映成另外$x$种字母使得其与$a$相同的位置尽可能多，令$w[i][j]$表示把$b$中的所有第$i$个字母变成第$j$个字母后，这些第$i$个字母所在位置与$a$对应位置相同的数量，那么每个位置$1\le i\le n$，对$w[b_i][a_i]$的贡献为$1$，以此建立二分图，问题转化为求该二分图的最大权匹配，时间复杂度$O(Tnm+26^3Tm)$

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=30;
int T,n,m,k,a[10005],b[10005];
char s[3];
int nx,ny;
int linker[maxn],lx[maxn],ly[maxn],slack[maxn];
int visx[maxn],visy[maxn],w[maxn][maxn];
int dfs(int x)
{
    visx[x]=1;
    for(int y=1;y<=ny;y++)
    {
        if(visy[y])
            continue;
        int tmp=lx[x]+ly[y]-w[x][y];
        if(tmp==0)
        {
            visy[y]=1;
            if(linker[y]==-1||dfs(linker[y]))
            {
                linker[y]=x;
                return 1;
            }
        }
        else if(slack[y]>tmp)
            slack[y]=tmp;
    }
    return 0;
}
int km()
{
    int i,j;
    memset(linker,-1,sizeof(linker));
    memset(ly,0,sizeof(ly));
    for(i=1;i<=nx;i++)  
        for(j=1,lx[i]=-INF;j<=ny;j++)
            if(w[i][j]>lx[i])
                lx[i]=w[i][j];
    for(int x=1;x<=nx;x++)
    {
        for(i=1;i<=ny;i++)
            slack[i]=INF;
        while(1)
        {
            memset(visx,0,sizeof(visx));
            memset(visy,0,sizeof(visy));
            if(dfs(x)) 
                break;  
            int d=INF;
            for(i=1;i<=ny;i++)
                if(!visy[i]&&d>slack[i])
                    d=slack[i];
            for(i=1;i<=nx;i++)
                if(visx[i])
                    lx[i]-=d;
            for(i=1;i<=ny;i++) 
                if(visy[i])
                    ly[i]+=d;
                else
                    slack[i]-=d;
        }
    }
    int res=0;
    for(i=1;i<=ny;i++)
        if(linker[i]!=-1)
            res+=w[linker[i]][i];
    return res;
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d%d",&n,&k,&m);
        for(int i=1;i<=n;i++)
        {
            scanf("%s",s);
            a[i]=s[0]-'A'+1;
        }
        while(m--)
        {
            for(int i=1;i<=n;i++)
            {
                scanf("%s",s);
                b[i]=s[0]-'A'+1;
            }
            nx=ny=26;
            memset(w,0,sizeof(w));
            for(int i=1;i<=n;i++)
                w[b[i]][a[i]]++;
            printf("%.4f\n",1.0*km()/n);
        }
    }
    return 0;
}