D. Serval and Rooted Tree(思维+树形dp+二分辅助)

最新推荐文章于 2022-06-11 09:14:36 发布

小菜鸡加油

最新推荐文章于 2022-06-11 09:14:36 发布

阅读量155

点赞数

分类专栏：思维 # 树形dp

本文链接：https://blog.csdn.net/zstuyyyyccccbbbb/article/details/115682675

版权

思维同时被 2 个专栏收录

514 篇文章 4 订阅

订阅专栏

树形dp

28 篇文章 0 订阅

订阅专栏

本文详细介绍了如何利用深度优先搜索（DFS）解决Codeforces上的问题1153D。首先解析了问题的思路，然后展示了一个C++代码实现，该代码包括关键的数据结构如向量、队列、映射和集合，以及一些基本算法如最小值查找。在DFS过程中，计算到达每个节点所需的最小大于等于给定值的个数。最后，通过二分查找确定满足条件的最小值。

摘要由CSDN通过智能技术生成

https://codeforces.com/problemset/problem/1153/D

思路：参考issue

https://issue-is-vegetable.blog.csdn.net/article/details/107760491?utm_medium=distribute.pc_relevant_t0.none-task-blog-2%7Edefault%7EBlogCommendFromBaidu%7Edefault-1.control&dist_request_id=&depth_1-utm_source=distribute.pc_relevant_t0.none-task-blog-2%7Edefault%7EBlogCommendFromBaidu%7Edefault-1.control

#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=3e5+1000;
typedef int LL;
inline LL read(){LL x=0,f=1;char ch=getchar();	while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
return x*f;}
LL dp[maxn],c[maxn],out[maxn];///dp[i]:到当前i这个节点要花费的最小的>=mid的个数
vector<LL>g[maxn];
void dfs(LL u,LL mid){
     dp[u]=0;///初始化
     if(g[u].size()==0){
        dp[u]=1;return;
     }
     LL minv=1e9;
     for(LL i=0;i<g[u].size();i++){
         LL v=g[u][i];
         dfs(v,mid);
         if(c[u]==1){
            minv=min(minv,dp[v]);
         }
         else if(c[u]==0){
            dp[u]+=dp[v];
         }
     }
     if(c[u]) dp[u]=minv;
}
int main(void){
   cin.tie(0);std::ios::sync_with_stdio(false);
   LL n;cin>>n;
   for(LL i=1;i<=n;i++) cin>>c[i];
   for(LL i=2;i<=n;i++){
       LL fa;cin>>fa;
       g[fa].push_back(i);
       out[fa]++;
   }
   LL k=0;
   for(LL i=1;i<=n;i++) if(out[i]==0) k++;
   LL l=1;LL r=n;
   while(l<r){
      LL mid=(l+r+1)>>1;
      ///memset(dp,0x3f,sizeof(dp));这样会在c[u]=0的时候dp[u]+=dp[v]有问题
      dfs(1,mid);
      if(k-mid+1>=dp[1]) l=mid;
      else r=mid-1;
   }
   cout<<l<<"\n";
   return 0;
}