D. Serval and Rooted Tree

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

原创最新推荐文章于 2022-06-11 09:14:36 发布 · 375 阅读

0 ·

CC 4.0 BY-SA版权

dfs 专栏收录该内容

33 篇文章

订阅专栏

本文解析了Codeforces比赛中的D题，题目要求在n个节点的树中，给叶子节点分配1~k的值，通过min和max操作求根节点的最大可能值。文章详细介绍了算法思路，包括使用DFS遍历树并记录每个节点能表示的叶子数量的最小值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

http://codeforces.com/contest/1153/problem/D

题意：n个节点的树，叶子可以任意分别给1~k（k为叶子个数）的值，每个节点有操作min，max分别取其子树的最小和最大的值；

问根节点1最大多少；

思路：每个节点记录它能表示的它叶子数量的最小的值（排序后最靠近前面的，第k大的数），min时就是只能是子树的叶子的数量了；max时就能是儿子最小的了（最靠前）；ans就是叶子数量减去根的值；

#include<algorithm>
#include<set>
#include<vector>
#include<queue>
#include<cmath>
#include<cstring>
#include<iostream>
#include<set>
#include<vector>
#include<queue>
#include<cmath>
#include<cstdio>
#include<map>
#include<stack>
#include<string>
using namespace std;

#define sfi(i) scanf("%d",&i)
#define pri(i) printf("%d\n",i)
#define sff(i) scanf("%lf",&i)
#define ll long long
#define mem(x) memset(x,0,sizeof(x))
#define INF 0x3f3f3f3f
#define eps 1e-16
#define PI acos(-1)
#define lowbit(x) ((x)&(-x))
#define zero(x) (((x)>0?(x):-(x))<eps)
#define fl() printf("flag\n")
#define MOD(x) ((x%mod)+mod)%mod
#define FASTIO ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
ll gcd(ll a,ll b){while(b^=a^=b^=a%=b);return a;}
const int maxn=3e5+9;
const int mod=1e9+7;

template <class T>
inline void sc(T &ret)
{
    char c;
    ret = 0;
    while ((c = getchar()) < '0' || c > '9');
    while (c >= '0' && c <= '9')
    {
        ret = ret * 10 + (c - '0'), c = getchar();
    }
}

ll power(ll x,ll n)
{
    ll ans=1;
    while(n)
    {
        if(n&1) ans=ans*x;
        n>>=1;
        x=x*x;
    }
    return ans;
}

vector<int>g[maxn];
bool op[maxn];
int k;
int dfs(int u)
{
    int num=0;
    if(op[u]) num=INF;
    if(g[u].size()==0) num=1,k++;
    for(int i=0;i<g[u].size();i++)
    {
        int v=g[u][i];
        int numv=dfs(v);

        if(op[u])
        {
            num=min(numv,num);
        }
        else num+=numv;
    }
    return num;
}
int main()
{
    FASTIO;
    int n;
    cin>>n;
    for(int i=1;i<=n;i++) cin>>op[i];
    for(int i=2;i<=n;i++)
    {
        int u;
        cin>>u;
        g[u].push_back(i);
    }
    int num=dfs(1);
    cout<<k-num+1<<endl;
    return 0;
}