谁来看一下我这里哪里错了
查看原帖
谁来看一下我这里哪里错了
376965
13006212285Pan楼主2023/8/22 10:40
#include <bits/stdc++.h>
using namespace std;
const int N=1e5+10;
map<int,int> other;
long long ans;
int main(){
	int n,k;
	cin >> n >> k;
	for(int i=1;i<=k;i++){
		int x;
		cin >> x;
		other[x]=i;
	}
	for(int i=1;i<n;i++){
		int a,b,c;
		cin >> a >> b >> c;
		if(other.find(a)!=other.end() && other.find(b)!=other.end()){
			ans+=c;
		}
	}
	cout << ans;
	return 0;
}

只有第一个点和第三个点AC了

2023/8/22 10:40
加载中...