www/min_tree_teacher.cpp

#include <cstdio>
#include <string>
#include <cstring>
#include <iostream>
#include<stdio.h>
#include <algorithm>
/*
4 5
1 2 2
1 3 2
1 4 3
2 3 4
3 4 3
*/
#define INT 0x3f3f3f3f
using namespace std;
const int maxn = 5010;
int a[maxn][maxn];//起始点和终点的数组
int vis[maxn],dist[maxn];//vis[]表示是否已经被访问，dist表示到当前节点的最小权值
int n,m;//节点数和边数
int u,v,w;//u到v的权值  
long long sum = 0;//最小生成树的权值
int prim(int pos) 
{
    dist[pos] = 0;//初始值设置为0
    for(int i = 1 ; i<=n ; i++)//遍历节点，使其默认为未访问
    {
        int cur = -1;//vis[-1]表示未被访问
        for(int j = 1; j <=n ; j++)
        {
            /*这个if用来判断当前是否是为距离已访问节点最近节点的权值为最小值
              cur = -1代表着未访问，那么肯定要访问，通过for循环来进行访问
              ，for（i）是将每个节点进行循环，for（j）是通过对cur和dist数值进行比值
              dist除1＝0，其他的初始化为INT，
            */
            if(!vis[j] && (cur == -1 || dist[j] < dist[cur]))
            cur = j;//更新访问状态
        }//第一次出门的时候cur = 1，j>n，
        if (dist[cur] >= INT) return INT;
            sum += dist[cur];
        vis[cur] = 1;
        for(int k = 1; k <= n; k++)
        /*第一次for已经选择了j节点作为起点，
        开始计算dist[]，后续的for（k）其实是更新min，*/
        {
        if(!vis[k]) 
        dist[k] = min(dist[k],a[cur][k]);
        }
        
    }
   return sum;
}
int main()
{
    scanf("%d%d",&n,&m);
    memset(a,0x3f,sizeof(a));
    memset(dist,0x3f,sizeof(dist));
    while(m--)
    {
        scanf("%d%d%d",&u,&v,&w);
        a[u][v] = min(a[u][v],w);
        a[v][u] = min(a[v][u],w);
    }
    int value = prim(1);
    if(value >= INT) puts("orz");
    else printf("%lld\n",sum);
    return 0;
}