安慰奶牛(最小生成树MST)

时间限制：1.0s   内存限制：256.0MB

问题描述

Farmer John变得非常懒，他不想再继续维护供奶牛之间供通行的道路。道路被用来连接N个牧场，牧场被连续地编号为1到N。每一个牧场都是一个奶牛的家。FJ计划除去P条道路中尽可能多的道路，但是还要保持牧场之间 的连通性。你首先要决定那些道路是需要保留的N-1条道路。第j条双向道路连接了牧场Sj和Ej(1 <= Sj <= N; 1 <= Ej <= N; Sj != Ej)，而且走完它需要Lj的时间。没有两个牧场是被一条以上的道路所连接。奶牛们非常伤心，因为她们的交通系统被削减了。你需要到每一个奶牛的住处去安慰她们。每次你到达第i个牧场的时候(即使你已经到过)，你必须花去Ci的时间和奶牛交谈。你每个晚上都会在同一个牧场(这是供你选择的)过夜，直到奶牛们都从悲伤中缓过神来。在早上 起来和晚上回去睡觉的时候，你都需要和在你睡觉的牧场的奶牛交谈一次。这样你才能完成你的 交谈任务。假设Farmer John采纳了你的建议，请计算出使所有奶牛都被安慰的最少时间。

输入格式

第1行包含两个整数N和P。

接下来N行，每行包含一个整数Ci。

接下来P行，每行包含三个整数Sj, Ej和Lj。

输出格式

输出一个整数, 所需要的总时间(包含和在你所在的牧场的奶牛的两次谈话时间)。

样例输入

5 6
10
10
20
6
30
1 2 5
2 3 5
2 4 12
3 4 17
2 5 15
3 5 6

样例输出

176

数据规模与约定

5 <= N <= 10000，N-1 <= P <= 100000，0 <= Lj <= 1000，1 <= Ci <= 1,000。

重新定义一下边权，选个权最小的节点开始走，跑个堆优化MST板子可A。(传统算法超时)

#include <bits/stdc++.h>
using namespace std;
#define MAX 150005
#define MOD 1000000007
typedef long long LL;

const int INF = 1e9+7;
const int VM = 10005;

struct node  //边的结构体
{
    int u, v, w;
};
node edge[VM * 10];
int rankt[VM];//分组的高度
int par[VM];//父节点
int talk[VM];
bool cmp(const node &a, const node &b)
{
    return a.w < b.w;//按w从小到大排序
}
int findx(int x)
{
    if (par[x] == x)//若根节点为本身
        return x;
    return par[x] = findx(par[x]);//路径压缩
}
bool same(int x, int y)  //判断为否在同一分组中
{
    return findx(x) == findx(y);
}
/**
void unite(int x, int y)
{
    x = findx(x);//查找根节点
    y = findx(y);//查找根节点
    if (x == y)//若以在同一分组
        return ;
    if (rankt[x] < rankt[y])//y所在分组的高度 大于x的
        par[x] = y;//将y作x的父节点。
    else
    {
        par[y] = x;//将x作为y的父节点
        if (rankt[x] == rankt[y])//若两个分组高度相同
            rankt[x]++;//x分组所在高度++
    }
}
**/
int main()
{
    int n, m,val=INF,i;
    scanf("%d %d", &n, &m);//获取边的个数 和顶点个数
    //int cnt = 1;//
    for ( i = 1; i <= n; i++){//初始化
        par[i] = i;
        scanf("%d",&talk[i]);
        val=min(val,talk[i]);
    }
    memset(rankt, 0, sizeof(rankt));//初始化
    for(i=1;i<=m;i++)
    {
        scanf("%d %d %d", &edge[i].u, &edge[i].v, &edge[i].w);//获取数据
        edge[i].w=2*edge[i].w+talk[edge[i].u]+talk[edge[i].v];
        //对于每条边，我们花费的时间是二倍的边权值加上两个端点的权值。然后用这个花费作为边权，求最小生成树。
        //cnt++;
    }
    sort(edge+1, edge + m+1, cmp);//按权值从小到大排序
    int ans = 0;//最小生成树 权值
    int countx = 0;//计数
    for ( i = 1; i <=m; i++)  //对所有的边
    {
        int u=findx(edge[i].u),v=findx(edge[i].v);
        if (u!=v)  //若两点不属于一个分组
        {
            ans += edge[i].w;//权值总和
            par[u] = v;
            //unite(e.u, e.v);//合并两点
            countx++;//计数
        }
        if (countx == n-1)break;
    }
    //输出
    //if (count == n-1)
        printf("%d\n", ans+val);
   // else
       // printf("?\n");
    return 0;
}