题目:
分析:
一开始,直接奔着并查集去了,结果打着打着,忽然发现,我们可以将所有连接敌军的道路暴力累加
然后就
A
C
AC
AC了
代码:
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<set>
#include<queue>
#include<vector>
#include<map>
#include<list>
#include<ctime>
#include<iomanip>
#include<string>
#include<bitset>
#include<deque>
#include<set>
#define LL long long
#define ch cheap
#define XJQ (int)%%%
using namespace std;
inline LL read() {
LL d=0,f=1;char s=getchar();
while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
while(s>='0'&&s<='9'){d=d*10+s-'0';s=getchar();}
return d*f;
}
int x[(int)1e5+5];
int main()
{
int n=read(),k=read();
int a;
for(int i=1;i<=k;i++) a=read()+1,x[a]=1;
LL ans=0;
for(int i=1;i<n;i++)
{
int a=read()+1,b=read()+1,w=read();
if(x[a]&&x[b]) ans+=w;
}
cout<<ans;
return 0;
}