-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathbutter.cpp
88 lines (80 loc) · 1.43 KB
/
butter.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
/*
ID:10179811
TASK:butter
LANG:C++
#FILE
*/
#include<cstdio>
#include<cstring>
#include<cstdlib>
using namespace std;
const int MAXP=805;
struct Edge
{
int x,y,z,next;
Edge(){}
Edge(int a,int b,int c):x(a),y(b),z(c){}
}edges[3000];
int first[MAXP],dis[MAXP],w[MAXP],Q[MAXP*MAXP],h,t,n,p,c,ans,tots;
bool flag[MAXP];
int min(int a,int b){return a<b?a:b;}
int spfa(int s)
{
memset(dis,0x3f,sizeof(dis));
memset(flag,true,sizeof(flag));
dis[s]=0;
h=t=0;
Q[t++]=s;
flag[s]=false;
while(h<t)
{
int x=Q[h++];
for(int i=first[x];i!=-1;i=edges[i].next)
if(dis[x]+edges[i].z<dis[edges[i].y])
{
dis[edges[i].y]=dis[x]+edges[i].z;
if(flag[edges[i].y])
{
Q[t++]=edges[i].y;
flag[edges[i].y]=false;
}
}
flag[x]=true;
}
int sum=0;
for(int i=1;i<=p;++i)sum+=dis[i]*w[i];
return sum;
}
void add(int a,int b,int c)
{
edges[tots]=Edge(a,b,c);
edges[tots].next=first[a];
first[a]=tots++;
}
int main()
{
freopen("butter.in","r",stdin);
freopen("butter.out","w",stdout);
scanf("%d %d %d",&n,&p,&c);
memset(w,0,sizeof(w));
memset(first,-1,sizeof(first));
while(n--)
{
int x;
scanf("%d",&x);
++w[x];
}
tots=0;
while(c--)
{
int u,v,c;
scanf("%d %d %d",&u,&v,&c);
add(u,v,c);
add(v,u,c);
}
ans=0x7fffffff;
for(int i=1;i<=p;++i)ans=min(ans,spfa(i));
printf("%d\n",ans);
//system("pause");
return 0;
}