标签：node int max Evil Book sec 怪物 dp out

[CF337D] Book of Evil - 树形dp

Description

有一棵树有 \(n\) 个节点，其中有 \(m\) 个节点发现了怪物。已知树上有一本魔法书，魔法书可以让到其距离小于等于 \(d\) 的点出现怪物，求魔法书所在点有几种可能。

Solution

对每个点，我们要求出三个值：子树内最远怪物，与最远不在同一个分支内的次远怪物（定义有点难以描述），子树外最远怪物（的距离）

#include <bits/stdc++.h>
using namespace std;

#define int long long
const int N = 1000005;

struct Node
{
    int max = INT_MIN;
    int sec = INT_MIN;
    int out = INT_MIN;
} node[N];

int n, m, d, a[N];

vector<int> g[N];

void dfs(int p, int from)
{
    if (a[p])
    {
        node[p].max = node[p].sec = 0;
    }
    for (int q : g[p])
        if (q != from)
        {
            dfs(q, p);
            if (node[p].max < node[q].max + 1)
            {
                node[p].sec = node[p].max;
                node[p].max = node[q].max + 1;
            }
            else
            {
                node[p].sec = max(node[p].sec, node[q].max + 1);
            }
        }
}

void dfsout(int p, int from)
{
    for (int q : g[p])
        if (q != from)
        {
            if (node[p].max == node[q].max + 1)
            {
                node[q].out = max(node[p].out + 1, node[p].sec + 1);
            }
            else
            {
                node[q].out = max(node[p].out + 1, node[p].max + 1);
            }
            dfsout(q, p);
        }
}

signed main()
{
    ios::sync_with_stdio(false);

    cin >> n >> m >> d;
    for (int i = 1; i <= m; i++)
    {
        int x;
        cin >> x;
        a[x] = 1;
    }

    for (int i = 1; i < n; i++)
    {
        int u, v;
        cin >> u >> v;
        g[u].push_back(v);
        g[v].push_back(u);
    }

    dfs(1, 0);
    dfsout(1, 0);

    int ans = 0;
    for (int i = 1; i <= n; i++)
    {
        int p = i;
        if (i == 1)
        {
            ans += node[p].max <= d && node[p].sec <= d;
        }
        else if (node[p].max <= d && node[p].out <= d)
        {
            ++ans;
        }
    }

    cout << ans << endl;
}

标签：node,int,max,Evil,Book,sec,怪物,dp,out
来源： https://www.cnblogs.com/mollnn/p/14390768.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。