ccf 最优灌溉

<div class="pdsec">问题描述</div><div class="pdcont">　　雷雷承包了很多片麦田，为了灌溉这些麦田，雷雷在第一个麦田挖了一口很深的水井，所有的麦田都从这口井来引水灌溉。
　　为了灌溉，雷雷需要建立一些水渠，以连接水井和麦田，雷雷也可以利用部分麦田作为“中转站”，利用水渠连接不同的麦田，这样只要一片麦田能被灌溉，则与其连接的麦田也能被灌溉。
　　现在雷雷知道哪些麦田之间可以建设水渠和建设每个水渠所需要的费用（注意不是所有麦田之间都可以建立水渠）。请问灌溉所有麦田最少需要多少费用来修建水渠。</div><div class="pdsec">输入格式</div><div class="pdcont">　　输入的第一行包含两个正整数n, m，分别表示麦田的片数和雷雷可以建立的水渠的数量。麦田使用1, 2, 3, ……依次标号。
　　接下来m行，每行包含三个整数a<sub>i</sub>, b<sub>i</sub>, c<sub>i</sub>，表示第a<sub>i</sub>片麦田与第b<sub>i</sub>片麦田之间可以建立一条水渠，所需要的费用为c<sub>i</sub>。</div><div class="pdsec">输出格式</div><div class="pdcont">　　输出一行，包含一个整数，表示灌溉所有麦田所需要的最小费用。</div><div class="pdsec">样例输入</div><div class="pddata">4 4
1 2 1
2 3 4
2 4 2
3 4 3</div><div class="pdsec">样例输出</div><div class="pddata">6</div>//最小生成树
#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;
int G[1001][1001];
int n,m,vis[1001],dis[1001];
int p(){
int cost=0;
vis[1]=1;dis[1]=0;
for(int i=1;i<=n;i++){
if(G[1][i]){
dis[i]=G[1][i];
}
}
for(int k=0;k<n-1;k++){
int s,min=0x3f3f3f3f;
for(int i=1;i<=n;i++){
if(min>dis[i]&&(!vis[i])){
min=dis[i];
s=i;
}
}
cost+=min;
vis[s]=1;
//	cout<<s<<' '<<min<<endl;
for(int i=1;i<=n;i++){
if(!vis[i]&&(dis[i]>G[s][i])){
dis[i]=G[s][i];
}
}
}
return cost;
}
int main(){
int a,b,c;
memset(vis,0,sizeof(vis));
memset(dis,0x3f3f3f3f,sizeof(dis));
memset(G,0x3f3f3f3f,sizeof(G));//此处应初始化为最大值
scanf("%d%d",&n,&m);
for(int i=0;i<m;i++){
scanf("%d%d%d",&a,&b,&c);
G[a][b]=c;
G[b][a]=c;
}
int ans=p();
cout<<ans<<endl;
return 0;
}