1090. Highest Price in Supply Chain (25)

就是一个树 dfs求得树的最大深度 和处于最大深度的叶子点数

#include <cstdio>
#include <algorithm>
#include <math.h>
#include <string.h>
#include <vector>
#include <map>
#include <iostream>
#define ll long long
#define INF 0x3f3f3f
using namespace std;
const int N=100000+500;
vector<int> v[N];
int maxcnt;
int ans[N];
void dfs(int x,int cnt)
{
    int i,to;
    maxcnt=max(cnt,maxcnt);
    ans[cnt]++;
    for(i=0;i<v[x].size();i++)
    {
        to=v[x][i];
        dfs(to,cnt+1);
    }
}
int main()
{
    int n,i,x,u;
    double p,r;
    maxcnt=-1;
    scanf("%d%lf%lf",&n,&p,&r);
    for(i=0;i<n;i++)
    {
        scanf("%d",&x);
        if(x==-1) u=n;
        else u=x;
        v[u].push_back(i);
    }
    dfs(n,0);
    for(i=0;i<maxcnt-1;i++)
        p*=(1.0+r*0.01);
    printf("%.2lf %d\n",p,ans[maxcnt]);
    return 0;
}