题目
(n=3e5
n个节点以1为根的一棵树,每个非叶子节点都有一个操作max或min(0表示min,1表示max),表示这个节点中的值应该分别等于其子节点中所有值的最大值或最小值。
假设树上有k个叶节点,你可以将每个叶节点填上[1,k]的数字,且每个数字只使用一次,求根节点的最大值
解答
思维题
f[i] 存一个点能取到的叶子节点的最小排名
op[i]==0 f[i]=min{f[y]}
op[i]==1 f[i]=sum{f[y]}
#include <iostream>
#include <cstring>
#include <vector>
using namespace std;
const int N=3e5+4;
vector<int> g[N];
int n,f[N],cnt[N],b[N];
const int inf=0x3f3f3f3f;
void dp(int u){
if(cnt[u]==0){
f[u]=1; return;
}
if(b[u]) f[u]=inf;
for(int i=0;i<g[u].size();i++){
int y=g[u][i]; dp(y);
if(b[u]) f[u]=min(f[u],f[y]);
else f[u]+=f[y];
}
}
int main(){
cin>>n;
int i,x,y,leaf=0;
for(i=1;i<=n;i++) cin>>b[i];
for(i=2;i<=n;i++)
cin>>x,g[x].push_back(i),cnt[x]++;
for(i=2;i<=n;i++) if(cnt[i]==0) leaf++;
dp(1);
cout<<leaf-f[1]+1;
}