Dessert Café

题目来源： 2020-2021 ACM-ICPC Asia Seoul Regional Contest

题目大意：

给出一棵n个点组成的树，边有权值，树上有k个关键点，现找出满足好的点个数有几个

若一个点$i$是一个好的点，它应该满足以下条件：

定义$d(x,y)$为点x和点y间的距离

对于某个关键点$z$，我们将$d(j,z)>d(i,z)$的点$j$加入到一个集合中（可以对所有关键点进行该操作）

如果最终得到的集合的大小为n-1（即除了i本身，其他点都被加入该集合了），则称点$i$为一个好的点

分析：

很容易想到，若一个点处于两个关键点的路径上，那它就是好的点

因此题目转化为：求有多少点处于两个关键点的路径上

将某个关键点作为树的根，以其为起点深搜，只要一个点的子树中含有关键点，则其位于两个关键点的路径上

代码：

#include<cstdio>
#include <map>
#include<algorithm>
#include <stack>
#include <vector>
#include <set>
#include <iostream>
using namespace std;

vector<int> G[100005];
int key[100005];
int vis[100005];
set<int>ans;

int dfs(int v){
   vis[v]=1;
   int flag=0;
   for(int i=0;i<G[v].size();i++){
       int t=G[v][i];
       if(vis[t]) continue;
       vis[t]=1;
       flag|=dfs(t);
   }
   if(flag)
        ans.insert(v);
   return key[v]|flag;
}

int main(){
    int n,k;
    scanf("%d%d",&n,&k);
    for(int i=0;i<n-1;i++){
         int a,b,c;
         scanf("%d%d%d",&a,&b,&c);
         G[a].push_back(b);
         G[b].push_back(a);
    }
    int temp;
    for(int i=0;i<k;i++){
         scanf("%d",&temp);
         key[temp]=1;
         ans.insert(temp);
    }
    dfs(temp);
    printf("%d\n",ans.size());
}