石油大 2020年秋季组队训练赛第二十五场 问题 C: Dessert Café （思维）

 

 

题意：给你n个点的树，边权为距离，然后给出k个特殊的点。现在要找的是点p，对于点p，有一个特殊点z，除了p和z之外的任何点x，d(p,z)<d(x,z)；而且又存在一个特殊的点z1，d(p,z1)<d(z,z1)。问这样的点p有多少个？

思路：边权没用，其实就是k个特殊点中，任意两点之间路径之间的个数（包含这k个特殊点）。因为是树，从一个特殊点开始dfs，看看其子树中有无特殊点即可。

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 1e6+10;
int n,k,a[N],ans;
vector<int> g[N];
bool vis[N],book[N];
int dfs(int u){
    if(vis[u]) return 0;
    bool flag=0;
    vis[u]=1;
    int len=g[u].size();
    for(int i=0;i<len;i++){
        int v=g[u][i];
        if(vis[v]) continue;
        if(dfs(v)&&!flag){
                if(!book[u]) ans++;
                flag=1;
        }
    }
    if(book[u]||flag) return 1; 
}
int main(void){
    scanf("%d%d",&n,&k);
    for(int i=1;i<n;i++){
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        g[u].push_back(v);
        g[v].push_back(u);      
    }   
    for(int i=1;i<=k;i++){
        scanf("%d",&a[i]);
        book[a[i]]=1;
    }
    ans=k;  
    dfs(a[1]);   
    cout<<ans<<endl;
    return 0;   
}