[Codeforces 277E][Round #170 div.1 E]Binary Tree on Plane

题目大意:平面上给出K个点,每个点可以向纵坐标比它小的点连边,边权为欧几里得距离,求一棵最小二叉生成树。
每个点只有一条入边,至多有两条出边,考虑使用最小费用最大流模型。
每个点拆成两个点i,i’,对于原图上的边u,v,u向v’连一条容量为1,费用为dis的边,源点向每个i连容量为2费用为0的边,每个i’向汇点连一条容量为1费用为0的边,最后检查除了最上面那个点的出边之外是否满流即可。

#include<ctime>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<cassert>
#include<cmath>
#include<vector>
#include<queue>
#include<stack>
#include<map>
#include<sstream>
#include<climits>
#define X first
#define Y second
#define DB double
#define lc now<<1
#define rc now<<1|1
#define MP make_pair
#define LL long long
#define pb push_back
#define sqr(_) ((_)*(_))
#define INF 0x3f3f3f3f
#define pii pair<int,int>
#define pdd pair<DB,DB>
#define ull unsigned LL
#define DEBUG(...) fprintf(stderr,__VA_ARGS__)
using namespace std;
template<typename T>void Read(T& x)
{
    x=0;int flag=0,sgn=1;char c;
    while(c=getchar())
    {
        if(c=='-')sgn=-1;
        else if(c>='0'&&c<='9')x*=10,x+=c-'0',flag=1;
        else if(flag)break;
    }
    x*=sgn;
}
const int MAXN=1000;
int n;
pdd pot[MAXN];
bool cmp(const pdd& x,const pdd& y)
{
    return x.Y==y.Y?x.X<y.X:x.Y<y.Y;
}
DB dis(const pdd& x,const pdd& y)
{
    return sqrt(sqr(x.X-y.X)+sqr(x.Y-y.Y));
}
struct Edge{
    int u,v,cap,flow;
    DB cost;
    Edge(int u=0,int v=0,int cap=0,int flow=0,DB cost=0):
        u(u),v(v),cap(cap),flow(flow),cost(cost){}
};
struct Mincost{
    vector<int> G[MAXN];
    vector<Edge> edges;
    int a[MAXN],p[MAXN],inq[MAXN],S,T;
    DB d[MAXN];
    void add(int u,int v,int c,DB co)
    {
        edges.pb(Edge(u,v,c,0,co));
        edges.pb(Edge(v,u,0,0,-co));
        int K=edges.size();
        G[u].pb(K-2);
        G[v].pb(K-1); 
    }
    bool SPFA(int& flow,DB& cost)
    {
        for(int i=0;i<MAXN;i++)d[i]=INF;
        memset(p,0,sizeof(p));
        memset(a,INF,sizeof(a)); 
        queue<int> q;
        q.push(S);
        inq[S]=1;
        d[S]=0;
        while(!q.empty())
        {
            int u=q.front();q.pop();
            inq[u]=0;
            for(int i=0;i<G[u].size();i++)
            {
                Edge& e=edges[G[u][i]];
                if(e.cap>e.flow&&d[e.v]>d[e.u]+e.cost)
                {
                    d[e.v]=d[e.u]+e.cost;
                    p[e.v]=G[u][i];
                    a[e.v]=min(a[u],e.cap-e.flow);
                    if(!inq[e.v])
                    {
                        q.push(e.v);
                        inq[e.v]=1;
                    }
                }
            }
        }
        if(d[T]==INF)return 0;
        flow+=a[T];
        cost+=(DB)a[T]*d[T];
        int U=T;
        while(U!=S)
        {
            edges[p[U]].flow+=a[T];
            edges[p[U]^1].flow-=a[T];
            U=edges[p[U]].u;
        }
        return 1;
    }
    DB mincost()
    {
        int flow=0;
        DB cost=0;
        while(SPFA(flow,cost));
        return cost;
    }
    bool check()
    {
        for(int i=0;i<edges.size();i++)
        {
            Edge& e=edges[i];
            if(e.u!=n+1&&e.v==T&&e.cap!=e.flow)
                return 0;
        }
        return 1;
    }
}Graph;
int main()
{
#ifndef ONLINE_JUDGE
    freopen("binary.in","r",stdin);
    freopen("binary.out","w",stdout);
#endif
    Read(n);
    for(int i=1;i<=n;i++)
        scanf("%lf %lf",&pot[i].X,&pot[i].Y); 
    sort(pot+1,pot+1+n,cmp);
    reverse(pot+1,pot+1+n);
    for(int i=1;i<=n;i++)
        for(int j=i+1;j<=n;j++)
            if(pot[i].Y>pot[j].Y)
                Graph.add(i,j+n,1,dis(pot[i],pot[j]));
    Graph.S=2*n+1,Graph.T=2*n+2;
    for(int i=1;i<=n;i++)
        Graph.add(Graph.S,i,2,0),
        Graph.add(i+n,Graph.T,1,0);
    DB ans=Graph.mincost();
    if(Graph.check())
        printf("%.10lf\n",ans);
    else
        cout<<-1<<endl;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值