poj3164-Command Network

给出平面上一些点，和连接它们的带权有向边，求把所有点连起来的最小总权值。

分析

由于这里边是有向的(unidirectional)，所以这是经典的最小树形图问题，可以说是最小树形图的模板题。

代码

这个写法是我乱想的。最近写图啊树啊都喜欢开一个结构体～

这个代码在poj上交，如果用g++的话会WA，因为奇怪的poj需要用%f输出浮点数。更奇怪的是c++直接编译错误，不想说了。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxn=202;
const double inf=1e100;
bool bian[maxn][maxn];
struct node {
    double x,y;
} a[maxn];
double dist(node &a,node &b) {
    return sqrt((a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y));
}
struct edge {
    int v;
    double w;
    int nxt;
};
struct graph {
    edge e[maxn*maxn];
    int h[maxn],tot,n,from[maxn],tic[maxn],tim,root,sta[maxn],top;
    double in[maxn],mi[maxn];
    bool vis[maxn],able[maxn];
    void clear() {
        memset(h,0,sizeof h);
        tot=0;
    }
    void add(int u,int v,double w) {
        e[++tot]=(edge){v,w,h[u]};
        h[u]=tot;
    }
    void dfs(int x) {
        vis[x]=true;
        for (int i=h[x],v=e[i].v;i;i=e[i].nxt,v=e[i].v) if (able[v]) {
            if (v!=root && in[v]>e[i].w) in[v]=e[i].w,from[v]=x;
            if (!vis[v]) dfs(v);
        }
    }
    bool circle() {
        memset(tic,0,sizeof tic),tim=0;
        for (int i=1;i<=n;++i) if (able[i]) {
            ++tim;
            top=0;
            for (int j=i;j;j=from[j]) if (!tic[j]) tic[j]=tim,sta[++top]=j; else if (tic[j]==tim) {
                for (int k=1;k<=n;++k) if (tic[k]==tim) tic[k]=0;
                do {
                    tic[sta[top--]]=tim;
                } while (sta[top+1]!=j);
                return true;
            }
        }
        return false;
    }
    bool work(double &ret) {
        memset(vis,0,sizeof vis);
        fill(in+1,in+n+1,inf);
        dfs(root);
        for (int i=1;i<=n;++i) if (able[i] && !vis[i]) {
            ret=-1;
            return false;
        }
        bool cc=circle();
        if (cc) {
            for (int i=1;i<=n;++i) if (able[i] && i!=root && tic[i]==tim) ret+=in[i];
        } else {
            for (int i=1;i<=n;++i) if (able[i] && i!=root) ret+=in[i];
        }
        return cc;
    }
    void reduce() {
        fill(mi+1,mi+n+1,inf);
        for (int i=1;i<=n;++i) if (able[i] && tic[i]!=tim) {
            double w=inf;
            for (int j=h[i],v=e[j].v;j;j=e[j].nxt,v=e[j].v) if (tic[v]==tim) w=min(w,e[j].w-in[v]);
            if (w!=inf) add(i,n+1,w);
        } 
        for (int i=1;i<=n;++i) if (able[i] && tic[i]==tim) for (int j=h[i],v=e[j].v;j;j=e[j].nxt,v=e[j].v) if (able[v] && tic[v]!=tim) mi[v]=min(mi[v],e[j].w);
        for (int i=1;i<=n;++i) if (able[i] && tic[i]!=tim && mi[i]!=inf) add(n+1,i,mi[i]); 
        for (int i=1;i<=n;++i) if (tic[i]==tim) able[i]=false;
        ++n;
        if (!able[root]) root=n;
    }
    double MGT(int _n) {
        memset(able,true,sizeof able),n=_n,root=1;
        double ret=0;
        while (work(ret)) {
            reduce();
        }   
        return ret;
    }
} A;
int main() {
#ifndef ONLINE_JUDGE
    freopen("test.in","r",stdin);
    freopen("my.out","w",stdout);
#endif
    int n,m;
    while (~scanf("%d%d",&n,&m)) {
        A.clear();
        memset(bian,0,sizeof bian);
        for (int i=1;i<=n;++i) scanf("%lf%lf",&a[i].x,&a[i].y);
        for (int i=1;i<=m;++i) {
            int u,v;
            scanf("%d%d",&u,&v);
            if (u==v || bian[u][v]) continue;
            bian[u][v]=true;
            double d=dist(a[u],a[v]);
            A.add(u,v,d);
        }
        double ans=A.MGT(n);
        if (ans<0) puts("poor snoopy"); else printf("%.2lf\n",ans);
    }
    return 0;
}

转载于:https://www.cnblogs.com/owenyu/p/6724593.html