挖沟

  胡队长带领HA实验的战士们玩真人CS，真人CS的地图由一些据点组成，现在胡队长已经占领了n个据点，为了方便，将他们编号为1-n，为了隐蔽，胡队长命令战士们在每个据点出挖一个坑，让战士们躲在坑里。由于需要在任意两个点之间传递信息，两个坑之间必须挖出至少一条通路，而挖沟是一件很麻烦的差事，所以胡队长希望挖出数量尽可能少的沟，使得任意两个据点之间有至少一条通路，顺便，尽可能的∑d[i][j]使最小（其中d[i][j]为据点i到j的距离）。

输入描述:

第一行有2个正整数n，m，m表示可供挖的沟数。
接下来m行，每行3个数a,b,v，每行描述一条可供挖的沟，该沟可以使a与b连通，长度为v。

输出描述:

输出一行，一个正整数，表示要使得任意两个据点之间有一条通路，至少需要挖长的沟。（数据保证有解）

示例1

输入

2 2
1 2 1
1 2 3

输出

1

示例2

输入

3 3
1 2 3
2 3 4
1 3 5

输出

7

备注:

对于100%的测试数据：
1 ≤ n ≤ 100000
1 ≤ m ≤ 500000
1 ≤ v ≤ 10000

//不可以用prim算法，否则会运行超时

#include <stdio.h>
#include <cstring> 
#include <vector> 
#include <algorithm>
using namespace std;
const int maxv=100010;
const int INF=1000000000;
struct node{
	int v;
	int dis;
	node(int _v,int _dis) : v(_v),dis(_dis){}
}; 
int n,m;
int d[maxv];
bool vis[maxv];
vector<node> Adj[maxv];
int prim(){
	fill(d,d+maxv,INF);
	memset(vis,0,sizeof(vis));
	d[1]=0;
	int ans=0;
	for(int i=0;i<n;i++){
		int u=-1,min=INF;
		for(int j=1;j<=n;j++){
			if(vis[j]==false&&d[j]<min){
				u=j;
				min=d[j];
			}
		}
		if(u==-1){
			return -1;
		}
		vis[u]=true;
		ans+=d[u];
		for(int j=0;j<Adj[u].size();j++){//不能从1到Adj[u].size(); 
			int v=Adj[u][j].v;
			if(vis[v]==false&&Adj[u][j].dis<d[v]){
				d[v]=Adj[u][j].dis;
			}
		}
	} 
	return ans;
}
int main(){
	int u,v,w;
	scanf("%d%d",&n,&m);
	for(int i=0;i<m;i++){
		scanf("%d%d%d",&u,&v,&w);
		Adj[u].push_back(node(v,w));
	}
	int ans=prim();
	printf("%d\n",ans);
	return 0;
}

注：kruskal可以通过；

#include <stdio.h>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxv=100010;//最大点数 
const int maxe=500010;//最大边数 
int father[maxv];
struct edge{
	int u,v;//两个端点 
	int cost;//边权 
}E[maxe];
int findFather(int x){
	int a=x;
	while(x!=father[x]){
		x=father[x];
	}
	while(a!=father[a]){//压缩路径 
		int z=a;
		a=father[a];
		father[z]=x;
	}
	return x;
}
bool cmp(edge a, edge b){
	return a.cost<b.cost;
}
int kruskal(int n,int m){//点数，边数 

	for(int i=1;i<=n;i++){//初始化 
		father[i]=i;
	}
	int ans=0,num_e=0;//边权和，边数
	sort(E,E+m,cmp); //按边数小的排序 
	for(int i=0;i<m;i++){
		int faU=findFather(E[i].u);
		int faV=findFather(E[i].v);
		if(faU!=faV){
			father[faU]=faV;
			ans+=E[i].cost;
			num_e++;
			if(num_e==n-1){
				break;
			}
		} 
	}
	if(num_e!=n-1){
		return -1;
	}else{
		return ans;
	}
}
int main(){
	int n,m;
	int u,v,w;
	scanf("%d%d",&n,&m);
	for(int i=0;i<m;i++){
		scanf("%d%d%d",&E[i].u,&E[i].v,&E[i].cost);//存图 
	}
	int ans=kruskal(n,m);
	printf("%d\n",ans);
	return 0;
}