POJ 3565 最小权匹配

题意:给你N个白点,N个黑点,将白点和黑点两两连起来,使得连起来之后的线段不相交。

思路:这题怎么看都像是计算几何,那么我们怎么把他转化称KM问题呢。

首先,这道题只有黑点和白点,这很容易让人想到匹配问题,那么为什么最佳的完备匹配,假设线段a - b , c- d ,相交,那么可以肯定dis(a,c) + dis(b,d) < dis(a ,b) + dis(c, d) 。

所以,最佳匹配的情况下,不会出现两条线段相交的情况。所以直接套用KM的最小权匹配算法就可以了 。

#include <set>
#include <map>
#include <stack>
#include <cmath>
#include <queue>
#include <cstdio>
#include <string>
#include <vector>
#include <iomanip>
#include <cstring>
#include <iostream>
#include <algorithm>
#define Max 2505
#define FI first
#define SE second
#define ll long long
#define PI acos(-1.0)
#define inf 0x3fffffff
#define LL(x) ( x << 1 )
#define bug puts("here")
#define PII pair<int,int>
#define RR(x) ( x << 1 | 1 )
#define mp(a,b) make_pair(a,b)
#define mem(a,b) memset(a,b,sizeof(a))
#define REP(i,s,t) for( int i = ( s ) ; i <= ( t ) ; ++ i )

using namespace std;
#define N 111
#define eps 1e-6
double Map[N][N] ;
double lx[N] , ly[N] ; //权
bool visx[N] , visy[N] ;
int linkx[N] , linky[N] ;
int n ;
int find(int now) { //最小权
    visx[now] = 1 ;
    for (int i = 1 ; i <= n ; i ++ ) {
        if(!visy[i] && Map[now][i] - lx[now] - ly[i] < eps) {//lx[i] + ly[j] <= w[i][j] 求最小权时
            visy[i] = 1 ;
            if(linky[i] == -1 || find(linky[i])) {
                linkx[now] = i ;
                linky[i] = now ;
                return 1 ;
            }
        }
    }
    return 0 ;
}
void KM() {
    mem(linkx ,-1) ;
    mem(linky ,-1) ;
    mem(ly ,0) ;
    for (int i = 1 ; i <= n ; i ++ ) {
        while(1) {
            mem(visx, 0) ;
            mem(visy, 0) ;
            if(find(i))break ;
            double d = inf ;
            for (int j = 1 ; j <= n ; j ++ )
                if(visx[j])
                    for (int k = 1 ; k <= n ; k ++ )
                        if(!visy[k])
                            d = min(d , Map[j][k] - lx[j] - ly[k]) ;//求出最小的差值
            for (int j = 1 ; j <= n ; j ++ ) {
                if(visx[j])lx[j] += d ;
                if(visy[j])ly[j] -= d ;
            }
        }
    }
}
struct kdq {
    ll x , y ;
} p1[N] ,p2[N] ;
double dis(int i ,int j) {
    return sqrt(1.0 * (p1[i].x - p2[j].x) * (p1[i].x - p2[j].x) + 1.0 * (p1[i].y - p2[j].y) * (p1[i].y - p2[j].y )) ;
}
int main() {
    while(cin >> n) {
        for (int i = 1 ; i <= n ; i ++ )scanf("%lld%lld",&p1[i].x ,&p1[i].y) ;
        for (int i = 1 ; i <= n ; i ++ )scanf("%lld%lld",&p2[i].x ,&p2[i].y) ;
        for (int i = 1 ; i <= n ; i ++ )lx[i] = inf ;
        for (int i = 1 ; i <= n ; i ++ )for (int j = 1 ; j <= n ; j ++ )Map[i][j] = dis(i , j) ,lx[i] = min(lx[i] , Map[i][j]) ;//求最小权,所以取边权最小值
        KM() ;
        for (int i = 1 ; i <= n ; i ++ ) {
            for (int j = 1 ; j <= n ; j ++ ) {
                if(linky[j] == i) {
                    cout << j << endl;
                    break ;
                }
            }
        }
    }
    return 0 ;
}


你可能感兴趣的:(POJ 3565 最小权匹配)