【PAT】Highest Price in Supply Chain (25)

题目链接

题解

本菜鸡刷题太少，虽然题目很简单，倒也是看了半天。题目意思大概如下：给出N个结点，P是单价，r是每次转手价格上升的百分比，然后第二行给出N个数（结点下标从0~N-1），每个数代表着第i个结点的供应商（也就是第i个结点的父亲，比如第一个数是1，就说明1是0的父亲，第二个数是5，就说明5是1的父亲），如果是-1，就代表那个点是根结点。

可以用DFS对叶子结点进行搜索，如果记录到了比最高单价还要高的价格，就进行更新，并记录最高单价的叶子数num=1；若计算出来的单价和最高单价一样，则num++。

代码

#include <iostream>
#include<bits/stdc++.h>
using namespace std;
const int maxn=100010;
int n;
double p,r,highprice=-1;
int deep=0,num=0;//deep用来记录深度，num记录最深的子结点个数
vector<int> Node[maxn];

void dfs(int root, int deep){
    if(Node[root].size()==0){   //寻找root的子结点
        double price=p*pow((1+r/100),deep);
        if(price>highprice){
            highprice=price;
            num=1;
        }
        else if(price==highprice){
            num++;
        }
    }
    deep++;
    for(int i=0;i<Node[root].size();i++){
        dfs(Node[root][i],deep);
    }

}

int main()
{
    int root;
    cin>>n>>p>>r;
    for(int i=0;i<n;i++){
        int temp;
        cin>>temp;
        if(temp==-1){
            root=i;
        }else{
            Node[temp].push_back(i);//temp为供应商，即父结点
        }
    }
    dfs(root,deep);
    printf("%.2lf %d",highprice,num);
    return 0;
}