Similarity 【HDU - 3718】【KM算法】

题目链接

---

  题意：

  有T次查询，每次给出一个长度为N的基础串，以及分别给出M个同样长度为N的待查询串，我们现在想知道，对应的交换顺序之后的使得它们可以成映像关系的最大的长度比上总长度的值为多少。

  思路：

  关于这道题的N是很大，但是由于K只有上限26，所以突破口在于这个K，我想让最多的匹配上，也就是匹配上的权值是最大的，这里该怎么办呢？于是就可以建立mp[27][27]，对应每个字符相互匹配得到的对应的贡献，这里的求贡献的复杂度可以使用O(N)的计算方式来求得。所以复杂度的重头戏依然被放在了最大权完美匹配这上面。这里的复杂度是的，因为在这里使用KM算法来做到实现最大权匹配问题。

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#include <unordered_map>
#include <unordered_set>
#define lowbit(x) ( x&(-x) )
#define pi 3.141592653589793
#define e 2.718281828459045
#define INF 0x3f3f3f3f
#define HalF (l + r)>>1
#define lsn rt<<1
#define rsn rt<<1|1
#define Lson lsn, l, mid
#define Rson rsn, mid+1, r
#define QL Lson, ql, qr
#define QR Rson, ql, qr
#define myself rt, l, r
#define MP(a, b) make_pair(a, b)
#define Min3(a, b, c) min(a, min(b, c))
using namespace std;
typedef unsigned long long ull;
typedef unsigned int uit;
typedef long long ll;
const int maxN = 27;
int N, M, K;
struct KM
{
    int n, mp[maxN][maxN], link_x[maxN], link_y[maxN];
    bool vis_x[maxN], vis_y[maxN];
    int que[maxN], top, fail, pre[maxN];
    ll hx[maxN], hy[maxN], slk[maxN];
    bool check(int i)
    {
        vis_x[i] = true;
        if(link_x[i])
        {
            que[fail++] = link_x[i];
            vis_y[link_x[i]] = true;
            return true;
        }
        while(i)
        {
            link_x[i] = pre[i];
            swap(i, link_y[pre[i]]);
        }
        return false;
    }
    void bfs(int S)
    {
        for(int i=1; i<=n; i++)
        {
            slk[i] = INF;
            vis_x[i] = vis_y[i] = false;
        }
        top = 0; fail = 1; que[0] = S;
        vis_y[S] = true;
        while(true)
        {
            ll d = 0;
            while(top < fail)
            {
                for(int i=1, j = que[top++]; i<=n; i++)
                {
                    if(!vis_x[i] && slk[i] >= (d = hx[i] + hy[j] - mp[i][j]))
                    {
                        pre[i] = j;
                        if(d) slk[i] = d;
                        else if(!check(i)) return;
                    }
                }
            }
            d = INF;
            for(int i=1; i<=n; i++)
            {
                if(!vis_x[i] && d > slk[i]) d = slk[i];
            }
            for(int i=1; i<=n; i++)
            {
                if(vis_x[i]) hx[i] += d;
                else slk[i] -= d;
                if(vis_y[i]) hy[i] -= d;
            }
            for(int i=1; i<=n; i++)
            {
                if(!vis_x[i] && !slk[i] && !check(i)) return;
            }
        }
    }
    void init()
    {
        for(int i=1; i<=n; i++)
        {
            link_x[i] = link_y[i] = 0;
            hy[i] = slk[i] = 0;
            vis_y[i] = false;
        }
        for(int i=1; i<=n; i++)
        {
            hx[i] = 0;
            for(int j=1; j<=n; j++)
            {
                if(hx[i] < mp[i][j]) hx[i] = mp[i][j];
            }
        }
    }
    void solve()
    {
        n = K;
        init();
        for(int i=1; i<=n; i++) bfs(i);
        ll ans = 0;
        for(int i=1; i<=n; i++) ans += hx[i] + hy[i];
        printf("%.4lf\n", 1. * ans / N);
    }
}km;
int Bas[10005], lsan[maxN], tot;
vector<int> B_pos[27], N_pos[27];
void do_with_nex_string()
{
    char ch[3];
    memset(lsan, 0, sizeof(lsan));
    for(int i=0; i<27; i++) N_pos[i].clear();
    tot = 0;
    for(int i=1; i<=N; i++)
    {
        scanf("%s", ch);
        Bas[i] = ch[0] - 'A';
        if(!lsan[Bas[i]])
        {
            lsan[Bas[i]] = ++tot;
        }
    }
    for(int i=1; i<=N; i++)
    {
        Bas[i] = lsan[Bas[i]];
        N_pos[Bas[i]].push_back(i);
    }
    for(int i=1; i<=K; i++) for(int j=0; j<=K; j++) km.mp[i][j] = 0;
    for(int i=1, len1, len2; i<=K; i++)
    {
        len1 = (int)B_pos[i].size();
        for(int j=1; j<=K; j++)
        {
            len2 = (int)N_pos[j].size();
            int l = 0, r = 0, sum = 0;
            while(l < len1 && r < len2)
            {
                if(B_pos[i][l] == N_pos[j][r])
                {
                    sum++;
                    l ++; r ++;
                }
                else if(B_pos[i][l] > N_pos[j][r]) r ++;
                else l++;
            }
            km.mp[i][j] = sum;
        }
    }
    km.solve();
}
int main()
{
    int T; scanf("%d", &T);
    while(T--)
    {
        scanf("%d%d%d", &N, &K, &M);
        for(int i=1; i<=K; i++) B_pos[i].clear();
        char ch[3];
        memset(lsan, 0, sizeof(lsan));
        tot = 0;
        for(int i=1; i<=N; i++)
        {
            scanf("%s", ch);
            Bas[i] = ch[0] - 'A';
            if(!lsan[Bas[i]])
            {
                lsan[Bas[i]] = ++tot;
            }
        }
        for(int i=1; i<=N; i++)
        {
            Bas[i] = lsan[Bas[i]];
            B_pos[Bas[i]].push_back(i);
        }
        for(int Cas=1; Cas <= M; Cas++)
        {
            do_with_nex_string();
        }
    }
    return 0;
}