洛谷 P2126 Mzc家中的男家丁

题目背景

ｍｚｃ与ｄｊｎ的…还没有众人皆知，所以我们要来宣传一下。

题目描述

ｍｚｃ家很有钱（开玩笑），他家有ｎ个男家丁，现在ｍｚｃ要将她们全都聚集起来（干什么就不知道了）。现在知道ｍｚｃ与男家丁们互相之间通信的时间，请算出把他们每个人叫到需要的总时间（要重复的哦）。保证能把他们每个人叫到。

输入输出格式

输入格式：

第一行有一个数n,表示有n个男家丁。第二行一个数m表示有m条通信路线。之后m行,每行三个数a[i],b[i],c[i],表示第a[i]个男家丁(或mzc)和第b[i]个男家丁(或mzc)通信需要时间(双向)。ai=0表示mzc。

输出格式：

一行，一个数ｓｕｍ，表示把他们每个人叫到需要的总时间。

输入输出样例

输入样例#1：

5
12
0 2 15
2 3 20
3 5 13
1 3 29
0 1 30
2 4 21
0 3 23
5 1 48
0 4 17
0 5 27
1 2 43
2 5 41

输出样例#1：

94

说明

ｎ＜＝2300

ｍ＜＝2300000

/*
明显的MST
但是这个题用kruskal就是WA(百思不得其姐)
所以就写了一遍prim 就过了
*/
#include<cstdio>
#include<cstring>
#include<iostream>
#define MAXN 10000
using namespace std;
int n,m,x,y,ans;
int f[2301][2301],l[MAXN];
bool used[MAXN];
inline void read(int&x) {
    x=0;int f=1;char c=getchar();
    while(c>'9'||c<'0') {if(c=='-') f=-1;c=getchar();}
    while(c>='0'&&c<='9') {x=(x<<1)+(x<<3)+c-48;c=getchar();}
    x=x*f;
}
inline void prim() {
    memset(l,127/3,sizeof l);
    memset(used,false,sizeof used);
    l[0]=0;
    int min,k;
    for(int i=0;i<=n;i++) {
        min=1e9;
        for(int j=0;j<=n;j++) {
            if(l[j]<min&&!used[j]) {
                min=l[j];
                k=j;
            }
        }
        used[k]=true;
        ans+=l[k];
        for(int j=1;j<=n;j++) {
            if(!used[j]&&f[k][j]<l[j])
              l[j]=f[k][j];
        }
    }
    return;
}
int main() {
    read(n);read(m);
    for(int i=0;i<=n;i++)
      for(int j=0;j<=n;j++)
        f[i][j]=1e9;
    for(int i=1;i<=m;i++) {
        read(x);read(y);
        read(f[x][y]);
        f[y][x]=f[x][y];
    }
    prim();
    printf("%d
",ans);
    return 0;
}