codeforces 338b Book of Evil

题意:树上有一个点可以覆盖距离离他d的所有点,告诉你已有多少点被覆盖,问这个点可能有几种情况


思路:从根开始搜索 构造一个数组记录深度比他深的 且已被覆盖  距离他最远的节点的距离 然后再进入搜索对于每个节点  拿到父节点离他最远的被覆盖节点的距离(不在这个节点下面) 即可求出答案


#include<cstring>
#include<cstdlib>
#include<iostream>
#include<stdio.h>
#include<cmath>
#include<algorithm>
#include<vector>
using namespace std;
#define N 100010
#define MOD 1000000009
#define LL long long

vector<int> e[N];
int dep[N];
int v[N];
int g[N];
int ans,n,m,k;


void dfs(int x,int fa)
{
    for(int i=0;i<e[x].size();i++)
    {
        int to=e[x][i];
        if(to==fa)continue;
        dfs(to,x);
        if(dep[to]>0||v[to])
            dep[x]=max(dep[x],dep[to]+1);
    }
}

void solve(int x,int fa,int dis)
{
    if(k>=max(dis,dep[x]))ans++;
    int l=0,r=0,a=0,b=0;
    for(int i=0;i<e[x].size();i++)
    {
        int to=e[x][i];
        if(to==fa)continue;
        g[to]=max(dis,l>0?l+1:a);
        l=max(l,dep[to]);
        a|=v[to];
    }

    for(int i=e[x].size()-1;i>=0;i--)
    {
        int to=e[x][i];
        if(to==fa)continue;
        g[to]=max(g[to],r>0?r+1:b);
        r=max(r,dep[to]);
        b|=v[to];
    }

    for(int i=0;i<e[x].size();i++)
    {
        int to=e[x][i];
        if(to==fa)continue;

        if(g[to]>0||v[x])
            solve(to,x,g[to]+1);
        else
            solve(to,x,0);
    }
}

int main()
{
    scanf("%d%d%d",&n,&m,&k);
    int x,y;
    for(int i=0;i<m;i++)
    {
        scanf("%d",&x);
        v[x]=1;
    }
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        e[x].push_back(y);
        e[y].push_back(x);
    }
    dfs(1,1);
    solve(1,1,0);
    printf("%d\n",ans);
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值