Cheering up the Cows,2008nov,usaco

那道題搞一搞會發現在選好的n-1 條邊上,每條邊走兩次,還需要花掉兩個端點需要的時間,才能回到原點。那麼我們不妨把每個邊權設成路長*2+兩個端點的時間,這樣建最小生成樹就可以了。還有,回到出發點還要加上出發點的值。

#include<stdio.h>
#include<algorithm>
using namespace std;
struct node
{
    int from;
    int to;
    int len;
}edge[2000000];
int cmp(node a,node b)
{
    return a.len<b.len?1:0;
}
int n,m;
int val[2000000];
int f[2000000];
int ans;
void in()
{
    for(int i=1;i<=n;i++)
    {
        f[i]=i;
    }
}
int find(int x)
{
    if(f[x]==x)
    {
        return x;
    }else
    {
        f[x]=find(f[x]);
        return f[x];
    }
}
void kru()
{
    int cnt=0;
    for(int i=1;i<=m;i++)
    {
        int fx=find(edge[i].from);
        int fy=find(edge[i].to);
        if(fx!=fy)
        {
            cnt++;
            f[fy]=fx;
            ans+=edge[i].len;
            if(cnt==n-1)
            {
                return ;
            }
             
        }
    }
}
int main()
{
     
    scanf("%d %d",&n,&m);
    int min=9999999;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&val[i]);
        if(val[i]<min)
        {
            min=val[i];
        }
    }
    for(int i=1;i<=m;i++)
    {
        int v;
        scanf("%d %d %d",&edge[i].from,&edge[i].to,&v);
        edge[i].len=val[edge[i].from]+val[edge[i].to]+v*2;
    }
    sort(edge+1,edge+1+m,cmp);
    in();
    kru();
    printf("%d",ans+min);
}


發佈了59 篇原創文章 · 獲贊 1 · 訪問量 3萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章