码迷,mamicode.com
首页 > 编程语言 > 详细

【算法】最小乘积生成树 & 最小乘积匹配 (HNOI2014画框)

时间:2018-04-07 13:55:25      阅读:200      评论:0      收藏:0      [点我收藏+]

标签:bubuko   continue   题目   bsp   com   答案   inf   画框   har   

今天考试的时候果然题目太难于是我就放弃了……转而学习了一下最小乘积生成树。

最小乘积生成树定义:

技术分享图片(摘自网上一篇博文)。

我们主要解决的问题就是当k = 2时,如何获得最小的权值乘积。我们注意到一张图可以有很多棵生成树,我们将每一棵生成树的权值记为(x, y),表示第一种权值之和为x, 第二种权值之和为y. 这样,很自然联想到二维平面上的坐标,每一棵生成树即为这个平面上的一个点。我们所想要寻找的点就是x * y最小的点。这样的点在什么位置?显然,若x1 <= x2, y1 <= y2,1号点的权值必然更小。所以我们的答案只可能处于这张平面图上的凸包的下凸壳上。

于是我们找到A,B两点,一个离y轴最近,一个离x轴最近,这两个点一定是下凸壳的两个端点。之后,我们再寻找到与AB距离最远的点C,用点C 更新答案后再以AC,BC为新的边向下递归求解。此时问题来了:如何找到这一个距离最远,且在AB下方的C点呢?我们将距离转化为面积,使用叉积求解。因为要求C点在AB下方,所以得到的叉积必为负数。又因为|叉积| = 四边形面积,所以得到的叉积必然是负的值中绝对值最大的那一个,即求解出与AB构成的叉积最小的C点。

然后就开始考虑式子的转化:min (B - A) * (C - A) = (B.x - A.x) (C.x - A.x) - (C.x - A.x) (B.y - A.y); 化开这个式子,省去常数部分,我们发现所求就是(A.y - B.y)* a[i][j] - (A.x - B.x)* b[i][j] 最小。我们考虑将这个东西看做权值,就可以用Kruskal求出使这个值最小的C点了。如果是匹配的话,则将i --> j 视作匹配的权值,将权值取反(因为要求求最小)后跑KM算法获得最大权值匹配。

下面的代码是仿照着的,但觉得写的很漂亮,放在这里大家可以参考一下。感谢原本的博主~

#include <bits/stdc++.h>
using namespace std;
#define maxn 1000
#define INF 99999999
int n, ans = INF, lx[maxn], ly[maxn], s[maxn], match[maxn];
int T, g[maxn][maxn], a[maxn][maxn], b[maxn][maxn];
bool visx[maxn], visy[maxn];

struct vec
{
    int x, y;
};

vec operator -(vec a, vec b)
{
    return (vec) { b.x - a.x, b.y - a.y };
}

int operator *(vec a, vec b)
{
    return a.x * b.y - a.y * b.x;
}

int read()
{
    int x = 0, k = 1;
    char c;
    c = getchar();
    while(c < 0 || c > 9) { if(c == -) k = -1; c = getchar(); }
    while(c >= 0 && c <= 9) x = x * 10 + c - 0, c = getchar();
    return x * k;
}

struct Graph
{
    void build(int wx, int wy)
    {
        for(int i = 1; i <= n; i ++)
            for(int j = 1; j <= n; j ++)
                g[i][j] = - (wx * a[i][j] + wy * b[i][j]);
    }    
    
    bool dfs(int u)
    {
        visx[u] = 1;
        for(int v = 1; v <= n; v ++)
        {
            if(visy[v]) continue;
            int tem = lx[u] + ly[v] - g[u][v];
            if(!tem)
            {
                visy[v] = 1;
                if(!match[v] || dfs(match[v]))
                {
                    match[v] = u;
                    return 1;
                }
            }
            else s[v] = min(s[v], tem);
        }
        return false;
    }
    
    vec KM()
    {
        memset(lx, 0, sizeof(lx)), memset(ly, 0, sizeof(ly));
        memset(match, 0, sizeof(match));
        for(int i = 1; i <= n; i ++)
            for(int j = 1; j <= n; j ++)
                lx[i] = max(lx[i], g[i][j]);
        for(int i = 1; i <= n; i ++)
        {
            memset(s, 63, sizeof(s));
            while(23333)
            {
                memset(visx, 0, sizeof(visx)), memset(visy, 0, sizeof(visy));
                if(dfs(i)) break;
                int tem = INF;
                for(int j = 1; j <= n; j ++)
                    if(!visy[j]) tem = min(tem, s[j]);
                for(int j = 1; j <= n; j ++)
                    if(visx[j]) lx[j] -= tem;
                for(int j = 1; j <= n; j ++)
                    if(visy[j]) ly[j] += tem;
                    else s[j] -= tem;
            }
        }
        vec re; re.x = 0, re.y = 0;
        for(int i = 1; i <= n; i ++)
            re.x += a[match[i]][i], re.y += b[match[i]][i];
        return re;
    }
}G;

void Solve(vec A, vec B)
{
    G.build(A.y - B.y, B.x - A.x);
    vec C = G.KM();
    ans = min(ans, C.x * C.y);
    if((A - B) * (A - C) >= 0) return;
    Solve(A, C), Solve(C, B);
}

int main()
{
    T = read();
    while(T --)
    {
        n = read();
        for(int i = 1; i <= n; i ++)
            for(int j = 1; j <= n; j ++)
                a[i][j] = read();
        for(int i = 1; i <= n; i ++)
            for(int j = 1; j <= n; j ++)
                b[i][j] = read();
        G.build(1, 0); 
        vec A = G.KM();
        G.build(0, 1);
        vec B = G.KM();
        ans = min(A.x * A.y, B.x * B.y);
        Solve(A, B);
        printf("%d\n", ans);
    }
    return 0;
} 

 

【算法】最小乘积生成树 & 最小乘积匹配 (HNOI2014画框)

标签:bubuko   continue   题目   bsp   com   答案   inf   画框   har   

原文地址:https://www.cnblogs.com/twilight-sx/p/8732993.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!