最小生成树

题目描述

如题，给出一个无向图，求出最小生成树，如果该图不连通，则输出orz

输入输出格式

输入格式：

第一行包含两个整数N、M，表示该图共有N个结点和M条无向边。（N<=5000，M<=200000）

接下来M行每行包含三个整数Xi、Yi、Zi，表示有一条长度为Zi的无向边连接结点Xi、Yi

输出格式：

输出包含一个数，即最小生成树的各边的长度之和；如果该图不连通则输出orz

输入输出样例

输入样例#1：  复制

4 5
1 2 2
1 3 2
1 4 3
2 3 4
3 4 3

输出样例#1：  复制

7

说明

时空限制：1000ms,128M

数据规模：

对于20%的数据：N<=5，M<=20

对于40%的数据：N<=50，M<=2500

对于70%的数据：N<=500，M<=10000

对于100%的数据：N<=5000，M<=200000

题目描述

如题，给出一个无向图，求出最小生成树，如果该图不连通，则输出orz

输入输出格式

输入格式：

第一行包含两个整数N、M，表示该图共有N个结点和M条无向边。（N<=5000，M<=200000）

接下来M行每行包含三个整数Xi、Yi、Zi，表示有一条长度为Zi的无向边连接结点Xi、Yi

输出格式：

输出包含一个数，即最小生成树的各边的长度之和；如果该图不连通则输出orz

输入输出样例

输入样例#1：  复制

4 5
1 2 2
1 3 2
1 4 3
2 3 4
3 4 3

输出样例#1：  复制

7

说明

时空限制：1000ms,128M

数据规模：

对于20%的数据：N<=5，M<=20

对于40%的数据：N<=50，M<=2500

对于70%的数据：N<=500，M<=10000

对于100%的数据：N<=5000，M<=200000

#include<bits/stdc++.h>
#define NUM 5050
#define maxint 0x7fffffff
int c[NUM][NUM];
void prim(int n)
{
int sum=0;
int lowcost[NUM];
int closet[NUM];
bool s[NUM]={false};
for(int i=1;i<=n;i++)
{
lowcost[i]=c[1][i];
closet[i]=1;
s[i]=false;
}
s[1]=true;
for(int i=1;i<n;i++)
{
int min =maxint;
int bj=1;
for(int j=2;j<=n;j++)
{
if(lowcost[j]<min&&(!s[j]))
{
min=lowcost[j];
bj=j;
}
}
sum+=min;
for(int j=2;j<=n;j++)
{
if((c[bj][j]<lowcost[j])&&(!s[j]))
{
lowcost[j]=c[bj][j];
closet[j]=bj;
}
}
s[bj]=true;
}
printf("%d",sum);
}


int main()
{



int n;
int m;
scanf("%d%d",&n,&m);
for(int i=0;i<=n;i++)
{
for(int j=0;j<=n;j++)
{
c[i][j]=maxint;

}
}
for(int i=1,u,v,w;i<=m;i++)
    {
        scanf("%d%d%d",&u,&v,&w);
        
        {
            c[u][v]=c[v][u]=w;
        }
    }//初始化cost数组
prim(n);
return 0;
}#include<iostream>
#include<cstring>
using namespace std;
#define NUM 5050
#define maxn 0x3f3f3f3f
int n,m,s;


int head[NUM];
int vis[NUM];
int clos[NUM];
struct edge{
int to;
int w;
int next;
}p[NUM];//前向星图存边
void prim()
{
for(int i=0;i<=n;i++)
clos[i]=maxn;
for(int i=head[1];i!=-1;i=p[i].next)
{
clos[p[i].to]=p[i].w;
}


clos[1]=0;


int sum=0;
vis[1]=1;
for(int i=1;i<n;i++)
{
int t=-1;
int min=maxn;
for(int j=1;j<=n;j++ )
{
if(!vis[j]&&min>clos[j])
{
min=clos[j];
t=j;
}
}
sum+=min;


vis[t]=1;
for(int j=head[t];j!=-1;j=p[j].next)
{
if(!vis[p[j].to])
{
if(clos[p[j].to]>p[j].w)
{
clos[p[j].to]=p[j].w;
}
}
}
}
cout<<sum;
}
int main()
{
memset(head,-1,sizeof(head));
cin>>n>>m;
for(int k=1;k<=m;k++)
{
int i,j,w;
cin>>i>>j>>w;
p[k].to=j;
p[k].w=w;
p[k].next=head[i];
head[i]=k;
}
prim();


return 0;
}