最小生成树 prime zoj1586

题意：在n个星球，每2个星球之间的联通需要依靠一个网络适配器，每个星球喜欢的网络适配器的价钱不同，先给你一个n，然后n个数，代表第i个星球喜爱的网络适配器的价钱，然后给出一个矩阵M[i][j]代表第i个星球到第j个星球联通所需的价钱,求联通所有星球所需的最小价钱

思路：prime 每次加入一条边的时候把边2个点的权值加进去即可 (zoj崩了，题也没法交啊，不知对错，啊~~~)

代码：

#include "iostream"
#include "string.h"
#include "stack"
#include "queue"
#include "string"
#include "vector"
#include "set"
#include "map"
#include "algorithm"
#include "stdio.h"
#include "math.h"
#define ll long long
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define mem(a) memset(a,0,sizeof(a))
using namespace std;
const int N=50000;
struct Edge{int fr,to;int w;friend bool operator< (Edge a, Edge b){return a.w>b.w;}
};
vector<Edge>Map[N];
int p[N];
void Prime(int n){int ans=0;Edge now;bool vis[N];mem(vis);priority_queue<Edge> Q;while(!Q.empty()) Q.pop();for(int i=0; i<Map[1].size(); ++i) Q.push(Map[1][i]);vis[1]=1;n--;while(n--){now=Q.top();Q.pop();if(vis[now.to])while(vis[now.to]){now=Q.top();Q.pop();}ans+=now.w;ans+=p[now.fr]+p[now.to];vis[now.to]=1;for(int i=0; i<Map[now.to].size(); ++i)if(!vis[Map[now.to][i].to]) Q.push(Map[now.to][i]);}printf("%d\n",ans);
}void Add(int u,int v,int w){Edge e;e.fr=u,e.to=v,e.w=w;Map[u].push_back(e);
}
int main(){int t,n,u,v,w;scanf("%d",&t);while(t--){scanf("%d",&n);mem(Map),mem(p);for(int i=1; i<=n; i++) scanf("%d",p+i);for(u=1; u<=n; ++u){for(v=1; v<=n; ++v){scanf("%d",&w);if(u!=v)Add(u,v,w),Add(v,u,w);//printf("%d %d %d\n",u,v,w);
            }}Prime(n);}return 0;
}