閱讀289 返回首頁    go 阿裏雲 go 技術社區[雲棲]


藍橋杯 算法訓練 安慰奶牛

 算法訓練 安慰奶牛  
時間限製:1.0s   內存限製:256.0M
問題描述

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
樣例輸出
178
數據規模與約定

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



思路:奶牛的牧場之間是一棵樹,會發現,對於每條邊,我們花費的時間是二倍的邊權值加上兩
個端點的權值。將每條邊(a, b)的權值Lj改變為2Lj+Ca+Cb,然後使用最小生成樹來計算。

至於選擇過夜的地方,根據題意,最後要多花費一個點權。找最小的即可

注意:藍橋杯給的測試數據是錯誤的,正確的請看上麵的測試數據!!

AC代碼:
#include<stdio.h>
#include<algorithm>
using namespace std;
struct node
{
   int a,b;
   int len;
}per[100010];
int cmp(node x,node y)
{
    if(x.len!=y.len) return x.len<y.len;
}
int Time[10010];
int main()
{
    int i,j,n,m,flag[10010],sum,num;
    scanf("%d %d",&n,&m);
    for(i=0;i<n;i++)
    {
       scanf("%d",&Time[i]);
    }
    for(i=0;i<m;i++)
    {
       scanf("%d%d%d",&per[i].a,&per[i].b,&per[i].len);   
       per[i].len=Time[per[i].a-1]+Time[per[i].b-1]+per[i].len*2;
    }
    sort(per,per+m,cmp);
    for(i=1;i<=n;i++) flag[i]=1;
       flag[per[0].a]=0;
       sum=0;
       for(i=0;i<m;i++)
       {
          num=flag[per[i].a]+flag[per[i].b];
          if(num==1)
          {
             sum+=per[i].len;
             flag[per[i].a]=flag[per[i].b]=0;
             i=0;
          }
       }
       sort(Time,Time+n);
       printf("%d\n",sum+Time[0]);
    return 0;
}


最後更新:2017-04-03 12:55:04

  上一篇:go Dell 1450用vs2010打開項目後,沒反應
  下一篇:go Dell 1450用vs2010打開項目後,沒反應