第十七周:( Sicily1083) Networking(c++)

来源:互联网 发布:c语言abs 是什么函数 编辑:程序博客网 时间:2024/06/18 16:17

原题链接:
http://soj.sysu.edu.cn/1083

思路:最小生成树。Prim算法求解。

代码:

//顶点编号为1-NumPoints//数据输入和初始化应该根据具体情况而定//Edges[Node2][Node1]=Edges[Node1][Node2]#include <iostream>using namespace std;#define INF 2000000000int Edges[105][105];int Post[55];bool Visited[55];int NumPoints,NumRoads;int find_min_node(){    int res=1,min=Post[1];    for(int i=1;i<=NumPoints;i++)        if(Post[i]<min){            min=Post[i];            res=i;        }    return res;}int Prim(){    int CostMinTree=0;    Visited[1]=true;    int add_node=1;    for(int i=2;i<=NumPoints;i++){        for(int j=1;j<=NumPoints;j++)            if((!Visited[j])&&(Edges[add_node][j]<Post[j]))                Post[j]=Edges[add_node][j];        add_node=find_min_node();        CostMinTree+=Post[add_node];        Visited[add_node]=true;        Post[add_node]=INF;    }    return CostMinTree;}int main(){    while(cin>>NumPoints){        if(NumPoints==0)            break;        cin>>NumRoads;        for(int i=1;i<=NumPoints;i++){            Visited[i]=false;            Post[i]=INF;            for(int j=1;j<=NumPoints;j++)                Edges[i][j]=INF;        }        for(int i=1;i<=NumRoads;i++){            int Node1,Node2,edge;            cin>>Node1>>Node2>>edge;            if(edge<Edges[Node1][Node2]){                Edges[Node1][Node2]=edge;                Edges[Node2][Node1]=edge;            }        }        cout<<Prim()<<endl;    }    return 0;}
原创粉丝点击