SSLOJ 1191.平津战役

版权声明：喜欢请点个大拇指，感谢各位dalao。弱弱说下，转载要出处呦 https://blog.csdn.net/qq_35786326/article/details/83113708

---

题目：

传送门

---

分析：

一开始，直接奔着并查集去了，结果打着打着，忽然发现，我们可以将所有连接敌军的道路暴力累加
然后就 $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;
}