PAT-1090 Highest Price in Supply Chain (25)

本题供销关系其实就是一颗树，即求树的深度和在该深度下叶节点数目。

利用C++容器vector来对数进行存储，然后DFS。

题目链接：https://www.patest.cn/contests/pat-a-practise/1090

#include <iostream>
#include <algorithm>
#include <vector>
#include <map>
#include <stack>
#include <queue>
#include <cmath>
using namespace std;
const int MAXSIZE=100005;

vector<int> NN[MAXSIZE];//NN[i][j]表示向i进货的第j个直接进货商 
int maxd = -1;//记录深度 
int count1 = 0;//到达最深层的次数 
//计算深度 
void DFS1(int root,int deep)
{
	maxd = max(deep,maxd); 
	for(int i=0;i<NN[root].size();++i)
	{
		DFS1(NN[root][i],deep+1);
	}
}
//计算到达最深层的次数
void DFS2(int root,int deep)
{
	if(deep == maxd)
		count1++;
	for(int i=0;i<NN[root].size();++i)
	{
		DFS2(NN[root][i],deep+1);
	}
}

int main(int argc, char** argv) {
	int i,j,k,n;
	int root;
	double p,rate;
	cin >> n >> p >> rate;
	for(i=0;i<n;++i)
	{
		cin >> k;
		if(k == -1)
			root = i;	
		else
			NN[k].push_back(i); 
	}
	DFS1(root,1); 
	DFS2(root,1);
	printf("%.2f %d\n",p*pow(1.0+rate*0.01,maxd-1),count1);
	//cout<<maxd<<" "<<count1<<endl;
	return 0;
}