zoukankan      html  css  js  c++  java
  • codeforces 337D Book of Evil(dp)

    转载请注明出处: http://www.cnblogs.com/fraud/           ——by fraud

    Book of Evil

    Paladin Manao caught the trail of the ancient Book of Evil in a swampy area. This area contains n settlements numbered from 1 to n. Moving through the swamp is very difficult, so people tramped exactly n - 1 paths. Each of these paths connects some pair of settlements and is bidirectional. Moreover, it is possible to reach any settlement from any other one by traversing one or several paths.

    The distance between two settlements is the minimum number of paths that have to be crossed to get from one settlement to the other one. Manao knows that the Book of Evil has got a damage range d. This means that if the Book of Evil is located in some settlement, its damage (for example, emergence of ghosts and werewolves) affects other settlements at distance d or less from the settlement where the Book resides.

    Manao has heard of m settlements affected by the Book of Evil. Their numbers are p1, p2, ..., pm. Note that the Book may be affecting other settlements as well, but this has not been detected yet. Manao wants to determine which settlements may contain the Book. Help him with this difficult task.

    Input

    The first line contains three space-separated integers n, m and d (1 ≤ m ≤ n ≤ 100000; 0 ≤ d ≤ n - 1). The second line contains m distinct space-separated integers p1, p2, ..., pm (1 ≤ pi ≤ n). Then n - 1 lines follow, each line describes a path made in the area. A path is described by a pair of space-separated integers ai and bi representing the ends of this path.

    Output

    Print a single number — the number of settlements that may contain the Book of Evil. It is possible that Manao received some controversial information and there is no settlement that may contain the Book. In such case, print 0.

    Sample test(s)
    Input
    6 2 3
    1 2
    1 5
    2 3
    3 4
    4 5
    5 6
    Output
    3
    Note

    Sample 1. The damage range of the Book of Evil equals 3 and its effects have been noticed in settlements 1 and 2. Thus, it can be in settlements 3, 4 or 5.

    题意

    给一棵n个结点的树,在树上的某个点上有一本"book of evil",在其周围的与其距离小于等于d的点都会受到其影响,已知m个受到影响的点,求有几个可能会有"book of evil"

    通过维护一个结点到其子树中的最长距离,以及除去存在最长距离之外,到另外子树的最长距离,然后再dfs一遍即可。

     1 #include <iostream>
     2 #include <cstdio>
     3 #include <cstring>
     4 #include <queue>
     5 #include <vector>
     6 using namespace std;
     7 #define MAXN 100010
     8 vector<int>G[MAXN];
     9 int dp[MAXN][2];
    10 int pre[MAXN][2];
    11 int ans=0;
    12 int d;
    13 void dfs(int u,int fa){
    14     for(int i=0;i<G[u].size();i++){
    15         int v=G[u][i];
    16         if(v==fa)continue;
    17         dfs(v,u);
    18         int d1=dp[v][0]+1;
    19         if(d1>dp[u][1]){
    20             dp[u][1]=d1;
    21             pre[u][1]=v;
    22             if(dp[u][1]>dp[u][0]){
    23                 swap(dp[u][0],dp[u][1]);
    24                 swap(pre[u][0],pre[u][1]);
    25             }
    26         }
    27     }
    28 }
    29 void dfs2(int u,int fa,int dis){
    30     if(dis>d)return;
    31     if(dp[u][0]<=d)ans++;//cout<<u<<endl;}
    32     int d1=max(dp[u][0],dis)+1;
    33     int d2=max(dp[u][1],dis)+1;
    34     for(int i=0;i<G[u].size();i++)
    35     {
    36         int v=G[u][i];
    37         if(v==fa)continue;
    38         if(pre[u][0]==v)dfs2(v,u,d2);
    39         else dfs2(v,u,d1);
    40     }
    41 }
    42 int main()
    43 {
    44     ios::sync_with_stdio(false);
    45     //freopen("in.in","r",stdin);
    46     int m,n;
    47     int u,v;
    48     cin>>n>>m>>d;
    49     for(int i=0;i<n;i++)dp[i][0]=dp[i][1]=-MAXN;
    50     for(int i=0;i<n;i++)pre[i][0]=pre[i][1]=-1;
    51     for(int i=0;i<m;i++){
    52         cin>>u;
    53         u--;
    54         dp[u][0]=dp[u][1]=0;
    55     }
    56     for(int i=0;i<n;i++)G[i].clear();
    57     for(int i=0;i<n-1;i++){
    58         cin>>u>>v;
    59         u--;v--;
    60         G[u].push_back(v);
    61         G[v].push_back(u);
    62     }
    63     ans=0;
    64     dfs(0,-1);
    65     dfs2(0,-1,-MAXN);
    66     cout<<ans<<endl;
    67     return 0;
    68 }
    代码君
  • 相关阅读:
    【业务自动化】iTop,全面支持ITIL流程的一款ITSM工具
    【Hadoop】HDFS源码解读
    【Hadoop】Hadoop RPC框架线程模型
    电商网站项目(一)用户模块---门户
    jdk1.5后枚举类的定义规则
    回顾static与final的作用
    @Springboot搭建项目controller层接收json格式的对象失败
    把一个文件中所有文件名或者文件路径读取到一个txt文件,然后在matlab中读取
    springboot用mybatis-generator自动生成mapper和model
    hibernate、mybatis、spring data 的对比
  • 原文地址:https://www.cnblogs.com/fraud/p/4338858.html
Copyright © 2011-2022 走看看