Cogs 309. [USACO 3.2] 香甜的黄油 dijkstra,堆,最短路,floyd

problem.php?pid=309

309. [USACO 3.2] 香甜的黄油

★★ 输入文件：butter.in 输出文件：butter.out 简单对比
时间限制：1 s 内存限制：128 MB

描述

农夫John发现做出全威斯康辛州最甜的黄油的方法：糖。把糖放在一片牧场上，他知道N（1<=N<=500）只奶牛会过来舔它，这样就能做出能卖好价钱的超甜黄油。当然，他将付出额外的费用在奶牛上。

农夫John很狡猾。像以前的Pavlov，他知道他可以训练这些奶牛，让它们在听到铃声时去一个特定的牧场。他打算将糖放在那里然后下午发出铃声，以至他可以在晚上挤奶。

农夫John知道每只奶牛都在各自喜欢的牧场（一个牧场不一定只有一头牛）。给出各头牛在的牧场和牧场间的路线，找出使所有牛到达的路程和最短的牧场（他将把糖放在那）

格式

INPUT FORMAT:

第一行: 三个数：奶牛数N，牧场数P（2<=P<=800），牧场间道路数C(1<=C<=1450)

第二行到第N+1行: 1到N头奶牛所在的牧场号

第N+2行到第N+C+1行：每行有三个数：相连的牧场A、B，两牧场间距离D（1<=D<=255），当然,连接是双向的

OUTPUT FORMAT:

一行输出奶牛必须行走的最小的距离和

SAMPLE INPUT

样例图形

         P2  
P1 @--1--@ C1
        |
        | 
      5  7  3
        |   
        |     C3
      C2 @--5--@
         P3    P4

SAMPLE OUTPUT

{说明：放在4号牧场最优 }

Pascal C C++

题解：

dijkstra+堆优化。（codevs上floyd也可以过，cogs上floyd我只能拿80分，但有人可以过。。。我写的太渣了。。。233）

从每个牛出发，把所有能到达的点全加上最短路的距离。最后找所有点中最小的即可。

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 #define INF 1e9
 4 struct node
 5 {
 6     int begin,end,value,next;
 7 }edge[2910];
 8 int cnt,Head[810],a[510],n,p,f[810],dis[810],Heap[810],pos[810],SIZE;
 9 int read()
10 {
11     int s=0,fh=1;char ch=getchar();
12     while(ch<'0'||ch>'9'){if(ch=='-')fh=-1;ch=getchar();}
13     while(ch>='0'&&ch<='9'){s=s*10+(ch-'0');ch=getchar();}
14     return s*fh;
15 }
16 void addedge(int bb,int ee,int vv)
17 {
18     edge[++cnt].begin=bb;edge[cnt].end=ee;edge[cnt].value=vv;edge[cnt].next=Head[bb];Head[bb]=cnt;
19 }
20 void addedge1(int bb,int ee,int vv)
21 {
22     addedge(bb,ee,vv);addedge(ee,bb,vv);
23 }
24 void Push1(int k)
25 {
26     int now=k,root;
27     while(now>1)
28     {
29         root=now/2;
30         if(dis[Heap[root]]<=dis[Heap[now]])return;
31         swap(Heap[root],Heap[now]);
32         swap(pos[Heap[root]],pos[Heap[now]]);
33         now=root;
34     }
35 }
36 void Insert(int k)
37 {
38     Heap[++SIZE]=k;pos[k]=SIZE;Push1(SIZE);
39 }
40 void Pop1(int k)
41 {
42     int now,root=k;
43     pos[Heap[k]]=0;Heap[k]=Heap[SIZE--];if(SIZE>0)pos[Heap[k]]=k;
44     while(root<=SIZE/2)
45     {
46         now=root*2;
47         if(now<SIZE&&dis[Heap[now+1]]<dis[Heap[now]])now++;
48         if(dis[Heap[root]]<=dis[Heap[now]])return;
49         swap(Heap[root],Heap[now]);
50         swap(pos[Heap[root]],pos[Heap[now]]);
51         root=now;
52     }
53 }
54 void dijkstra(int start)
55 {
56     int i,u,v;
57     for(i=1;i<=p;i++)dis[i]=INF;dis[start]=0;
58     for(i=1;i<=p;i++)Insert(i);
59     while(SIZE>0)
60     {
61         u=Heap[1];Pop1(pos[u]);
62         for(i=Head[u];i!=-1;i=edge[i].next)
63         {
64             v=edge[i].end;
65             if(dis[v]>dis[u]+edge[i].value){dis[v]=dis[u]+edge[i].value;Push1(pos[v]);}
66         }
67     }
68 }
69 int main()
70 {
71     freopen("butter.in","r",stdin);
72     freopen("butter.out","w",stdout);
73     int c,s1,s2,s3,i,j,MN;
74     n=read();p=read();c=read();
75     for(i=1;i<=n;i++)a[i]=read();
76     memset(Head,-1,sizeof(Head));cnt=1;
77     SIZE=0;
78     for(i=1;i<=c;i++)
79     {
80         s1=read();s2=read();s3=read();
81         addedge1(s1,s2,s3);
82     }
83     memset(f,0,sizeof(f));
84     for(i=1;i<=n;i++)
85     {
86         dijkstra(a[i]);
87         for(j=1;j<=p;j++)f[j]+=dis[j];
88     }
89     MN=INF;
90     for(i=1;i<=p;i++)MN=min(MN,f[i]);
91     printf("%d",MN);
92     fclose(stdin);
93     fclose(stdout);
94     return 0;
95 }