您的位置:首页 > 其它

最小生成树模板(Prim+优先队列)

2012-05-11 10:48 344 查看
const int INF = 0xfffffff;
const int M = 1005;
Graph<M,M*M> g;
int Dis[M];
bool Vist[M];

struct Node{
Node(){}
Node(int Len, int Ind):len(Len),ind(Ind){}
int len,ind;
bool operator<(const Node& node) const{
return len > node.len;
}
};

int Prim(int s, int n)
{
fill(Dis,Dis+M,INF);
int result = 0;
priority_queue<Node> Q;
CLR(Vist,false);
Dis[s] = 0;
struct Node cur;
Q.push(Node(0,s));
while(!Q.empty())
{
cur = Q.top(); Q.pop();
if(cur.len != Dis[ cur.ind ]) continue;
Vist[ cur.ind ] = true;
result += cur.len;
for(int i = g.H[ cur.ind ]; i != -1; i = g.Next[i])
{
if(!Vist[ g.Num[i] ] && g.Len[i] < Dis[ g.Num[i] ])
{
Dis[ g.Num[i] ] = g.Len[i];
Q.push(Node(Dis[ g.Num[i] ],g.Num[i]));
}
}
}
return result;
}
须加邻接表模板,头文件queue.h,宏定义#define CLR(arr,v) memset(arr,v,sizeof(arr))
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: