Codeforce 337D(树的直径变形)

最新推荐文章于 2021-05-01 00:31:05 发布

原创最新推荐文章于 2021-05-01 00:31:05 发布 · 786 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#acm #codeforce #动态规划

动态规划同时被 2 个专栏收录

226 篇文章

订阅专栏

---------树形dp

33 篇文章

订阅专栏

本文介绍了一种算法，用于解决在一个给定的树结构中寻找距离指定节点集距离不超过给定阈值的节点数量的问题。通过两次深度优先搜索（DFS）实现，首次搜索确定每个指定节点的最长路径，第二次则计算所有符合条件的节点。

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

链接：点击打开链接

题意：在一个n个点的树中，给出m个点，问这颗树中距离这m个点的距离小于等于d的点的个数

代码：

#include <vector>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
const int INF=0x3f3f3f3f;
struct node{
    int to,cost;
    node(int x,int y){
        to=x,cost=y;
    }
};
vector<node> G[100005];
int n,dp[100005],vis[100005],dis[100005][5],sign[100005];
void dfs1(int s){
    int i,tmp;
    sign[s]=1;
    for(i=0;i<G[s].size();i++){
        tmp=G[s][i].to;
        if(!sign[tmp]){
        dfs1(tmp);
        if(dis[tmp][0]+G[s][i].cost>dis[s][0]){
            dis[s][0]=dis[tmp][0]+G[s][i].cost;
            vis[s]=tmp;
        }
        }
    }
    for(i=0;i<G[s].size();i++){
        tmp=G[s][i].to;
        if(tmp!=vis[s])
        dis[s][1]=max(dis[s][1],dis[tmp][0]+G[s][i].cost);
    }
}
void dfs2(int s){
    int i,tmp;
    sign[s]=1;
    for(i=0;i<G[s].size();i++){
        tmp=G[s][i].to;
        if(!sign[tmp]){
        if(tmp==vis[s])
        dp[tmp]=max(dp[s],dis[s][1])+G[s][i].cost;
        else
        dp[tmp]=max(dp[s],dis[s][0])+G[s][i].cost;
        dfs2(tmp);
        }
    }
}
int main(){                                     //就是树的直径变形，有原来向上和向下
    int i,j,x,y,m,d,ans,num;                    //的最长长度变为求这个点到给定的m个点中
    while(scanf("%d%d%d",&n,&m,&d)!=EOF){       //的最长向上和向下的距离
        memset(vis,0,sizeof(vis));
        memset(sign,0,sizeof(sign));            //可以先看树的直径
        for(i=1;i<=n;i++){                      //http://blog.youkuaiyun.com/stay_accept/article/details/50813783
        G[i].clear();
        dp[i]=dis[i][0]=dis[i][1]=-INF;
        }
        for(i=1;i<=m;i++){
            scanf("%d",&num);
            dp[num]=dis[num][0]=dis[num][1]=0;
        }
        for(i=2;i<=n;i++){
            scanf("%d%d",&x,&y);
            G[x].push_back(node(y,1));
            G[y].push_back(node(x,1));
        }
        dfs1(1);
        memset(sign,0,sizeof(sign));
        dfs2(1);
        ans=0;
        for(i=1;i<=n;i++)
        if(dis[i][0]<=d&&dp[i]<=d)
        ans++;
        printf("%d\n",ans);
    }
    return 0;
}