poj 3565 Ants

题目

平面上共有2*n个点，n个是白点，n个是黑点。对于每个白点，找到一个黑点，用线段连接起来。要求所有的线段不相交。

分析

在心理课上看到的题
一开始毫无头绪，后面想了一下，发现如果不相交，其实等价于让每条线段的长度之和最小。

来幅图

于是就变成了最小费用流。

code

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<queue>

using namespace std;

const int maxm=50000+10,maxn=800+10;
struct arr{
    int x,y;
    double w;
    int c;
    int next;
}edge[maxm*3];
int ls[maxm*3];
int f[maxm*3];
double dis[maxn*2];
int pro[maxn*2];
int edge_m=1;

int n,m;

void add(int x,int y,double w,int c)
{
    edge_m++;
    edge[edge_m]=(arr){x,y,w,c,ls[x]},f[edge_m]=c,ls[x]=edge_m;
    edge_m++;
    edge[edge_m]=(arr){y,x,-w,-c,ls[y]},f[edge_m]=0,ls[y]=edge_m;
}

int v[maxn*2];
int s,t;

int spfa()
{
    queue<int> q;
    for (int i=0;i<=t;i++) dis[i]=2000000000;
    memset(v,0,sizeof(v));
    memset(pro,0,sizeof(pro));
    dis[s]=0;
    q.push(s); v[s]=1;
    do{
        int x=q.front();
        q.pop();
        for (int i=ls[x];i;i=edge[i].next)
        {
            if ((f[i])&&(dis[edge[i].y]>dis[x]+edge[i].w))
            {
                dis[edge[i].y]=dis[x]+edge[i].w;
                pro[edge[i].y]=i;
                if (!v[edge[i].y])
                {
                    v[edge[i].y]=1;
                    q.push(edge[i].y);
                }
            }
        }
        v[x]=0;
    }while(!q.empty());
    if (dis[t]==dis[0]) return 0;
    else return 1;
}

int ans[maxn];

void mcf()
{
    int mi=200000000;
    int x=t;
    while (pro[x]!=0)
    {
        mi=min(f[pro[x]],mi);
        ans[edge[pro[x]].x]=x;
        x=edge[pro[x]].x;
    }
    x=t;
    while (pro[x]!=0)
    {
        f[pro[x]]-=mi;
        f[pro[x]^1]+=mi;
        x=edge[pro[x]].x;
    }
}

int work()
{
    while (spfa())
    {
        mcf();
    }
} 

int x[maxn],y[maxn];

double diss(int i,int j)
{
    return sqrt((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]));
}

int main()
{
    scanf("%d",&n);
    edge_m=1;
    for (int i=1;i<=2*n;i++)
    {
        scanf("%d%d",&x[i],&y[i]);
    }
    for (int i=1;i<=n;i++)
    {
        for (int j=n+1;j<=n*2;j++)
            add(i,j,diss(i,j),1);
    }
    s=2*n+1;
    t=2*n+2;
    for (int i=1;i<=n;i++)
        add(s,i,0,1);
    for (int i=n+1;i<=2*n;i++)
        add(i,t,0,1);
    work();
    for (int i=1;i<=n;i++)
    {
        printf("%d\n",ans[i]-n);
    }
    return 0;
}