POJ 3164 Command Network 最小树形图模板

最小树形图求的是有向图的最小生成树,跟无向图求最小生成树有很大的区别。

步骤大致如下:

1.求除了根节点以外每个节点的最小入边,记录前驱

2.判断除了根节点,是否每个节点都有入边,如果存在没有入边的点,说明树形图不存在,退出。

3.沿着节点的前驱找,如果发现环,把环缩点。

4.如果不存在环,结束。否则,跳到1.

 

其实我是来贴模板的。。。

比较好的讲解:最小树型图的求解与实现

代码讲解比较好的:hdu4009 Transfer water ( 最小树形图的模板 )

用IO优化时死活TLE,改了scanf就过了,后来用!=EOF发现数据不全,怪不得TLE了,囧

 

 

#include <set>

#include <map>

#include <list>

#include <cmath>

#include <queue>

#include <stack>

#include <string>

#include <vector>

#include <cstdio>

#include <cstring>

#include <iostream>

#include <algorithm>



using namespace std;



typedef long long ll;

typedef unsigned long long ull;



#define debug puts("here")

#define rep(i,n) for(int i=0;i<n;i++)

#define rep1(i,n) for(int i=1;i<=n;i++)

#define REP(i,a,b) for(int i=a;i<=b;i++)

#define foreach(i,vec) for(unsigned i=0;i<vec.size();i++)

#define pb push_back

#define RD(n) scanf("%d",&n)

#define RD2(x,y) scanf("%d%d",&x,&y)

#define RD3(x,y,z) scanf("%d%d%d",&x,&y,&z)

#define RD4(x,y,z,w) scanf("%d%d%d%d",&x,&y,&z,&w)

#define All(vec) vec.begin(),vec.end()

#define MP make_pair

#define PII pair<int,int>

#define PQ priority_queue

#define cmax(x,y) x = max(x,y)

#define cmin(x,y) x = min(x,y)

#define Clear(x) memset(x,0,sizeof(x))

/*



#pragma comment(linker, "/STACK:1024000000,1024000000")



int size = 256 << 20; // 256MB

char *p = (char*)malloc(size) + size;

__asm__("movl %0, %%esp\n" :: "r"(p) );



*/



char IN;

bool NEG;

inline void Int(int &x){

    NEG = 0;

    while(!isdigit(IN=getchar()))

        if(IN=='-')NEG = 1;

    x = IN-'0';

    while(isdigit(IN=getchar()))

        x = x*10+IN-'0';

    if(NEG)x = -x;

}

inline void LL(ll &x){

    NEG = 0;

    while(!isdigit(IN=getchar()))

        if(IN=='-')NEG = 1;

    x = IN-'0';

    while(isdigit(IN=getchar()))

        x = x*10+IN-'0';

    if(NEG)x = -x;

}



/******** program ********************/



const int MAXN = 105;

const double INF = 1e50;



int pre[MAXN],id[MAXN],use[MAXN];

int px[MAXN],py[MAXN];

double in[MAXN];



struct Edge{

    int x,y;

    double cost;

    Edge(){}

    Edge(int _x,int _y,double _cost):x(_x),y(_y),cost(_cost){}

}g[MAXN*MAXN];

double treeDiagrm(int root,int n,int m){

    double ans = 0;

    while(true){

        rep1(i,n)

            in[i] = INF;



        rep1(i,m){

            int x = g[i].x , y = g[i].y;

            double cost = g[i].cost;

            if(cost<in[y]&&y!=x)

                pre[y] = x , in[y] = cost;

        }

        in[root] = 0 , pre[root] = root;

        rep1(x,n){

            if(in[x]>1e20)return -1;

            ans += in[x];

        }

        Clear(use);

        Clear(id);

        int tot = 0;

        rep1(i,n){

            if(use[i])continue;

            int x = i;

            while(!use[x])use[x] = i , x = pre[x];

            if(use[x]!=i||x==root)continue;

            id[x] = ++ tot;

            for(int t=pre[x];t!=x;t=pre[t])

                id[t] = tot;

        }

        if(!tot)break;

        rep1(i,n)if(!id[i])id[i] = ++ tot;



        rep1(i,m){

            g[i].cost -= in[g[i].y];

            g[i].x = id[g[i].x];

            g[i].y = id[g[i].y];

        }

        n = tot;

        root = id[root];

    }

    return ans;

}



double cal(int x,int y){

    return sqrt(0.0+x*x+y*y);

}



int main(){



#ifndef ONLINE_JUDGE

    freopen("sum.in","r",stdin);

    //freopen("sum.out","w",stdout);

#endif



    int x,y,n,m;

    while(~RD2(n,m)){

        rep1(i,n)

            RD2(px[i],py[i]);

        rep1(i,m){

            RD2(x,y);

            g[i] = Edge(x,y,cal(px[x]-px[y],py[x]-py[y]));

        }

        double ans = treeDiagrm(1,n,m);

        if(ans>0)

            printf("%.2lf\n",ans);

        else puts("poor snoopy");

    }



    return 0;

}

 

  

 

 

 

你可能感兴趣的:(command)