cf 337 D（dfs）

最新推荐文章于 2024-08-15 18:30:00 发布

原创最新推荐文章于 2024-08-15 18:30:00 发布 · 1.3k 阅读

0 ·

CC 4.0 BY-SA版权

acm 同时被 3 个专栏收录

71 篇文章

订阅专栏

45 篇文章

订阅专栏

搜索

32 篇文章

订阅专栏

本文提供了一种解决 CodeForces 平台上的 337D 问题的方法。通过找到被攻击的最远两点，并检查其他点到这两点的距离是否均小于给定值 d 来判断解的可行性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

链接：http://codeforces.com/problemset/problem/337/D

找到已经被攻击的最远距离的两个点，再遍历其他点到这两点之间的距离，当两点之间的距离都小于d，这符合题意

#include <bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
int n,m,d;
int a[maxn],dist1[maxn],dist2[maxn],dist3[maxn];
map<int,int>ma;
map<int,int>mark;
vector<int>g[maxn];

void dfs(int v,int cnt,int dist[])
{
    if(mark[v])return;
    mark[v]++;
    dist[v]=cnt;
    for(int i=0;i<g[v].size();i++)
        dfs(g[v][i],cnt+1,dist);
}

int main()
{
    scanf("%d%d%d",&n,&m,&d);
    for(int i=1;i<=m;i++)
    {
        scanf("%d",a+i);
        ma[a[i]]++;
    }
    for(int i=1;i<n;i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        g[u].push_back(v);
        g[v].push_back(u);
    }
    mark.clear();
    dfs(1,0,dist1);
    int u=0,v=0,ans=0,mm=0;
    for(int i=1;i<=n;i++)
        if(ma[i]&&mm<dist1[i])
            mm=dist1[i],u=i;
    //cout<<u<<endl;
    mark.clear();
    mm=0;
    dfs(u,0,dist2);
    for(int i=1;i<=n;i++)
        if(ma[i]&&mm<dist2[i])
            mm=dist2[i],v=i;
  //  cout<<v<<endl;
    mark.clear();
    dfs(v,0,dist3);
    for(int i=1;i<=n;i++)
        if(dist2[i]<=d&&dist3[i]<=d)
            ans++;
    printf("%d\n",ans);
    return 0;
}