题目大意:
给出一颗销售供应的树,树根唯一。在树根处货物的价格为P,然后开始从根节点每走一层,该层的货物价格在父亲节点的价格上增加r%,给出每个叶节点的货物量,求他们的价格之和
思路:
1.类似[PAT-A 1090]Highest Price in Supply Chain,在1090的基础上增加货物量的设定,增加点权
2.节点的结构体设置
struct node {
double data;
vector<int> child;
}Node[maxn];
3.设置当前访问节点index与当前深度depth
注意:
1.递归边界。当index的子节点个数为0,表示达到了叶节点,计算价格pow(1+r,depth)*Node[i].data.
2.堆节点index的所有子节点进行递归,令depth+1
3.计算深度时,根节点的深度应该记为0.
4.输入时Ki如果为0,表示该节点为叶节点,后面跟的为叶节点的权值。如果ki不为0,表示该节点不是叶节点,后面跟的节点为该节点的左右孩子。
AC代码:
//PAT_A 1079
#include<cstdio>
#include<cmath>
#include<vector>
using namespace std;
const int maxn = 100010;
struct node{
double data;
vector<int> child;
}Node[maxn];
int n;
double p, r, ans;
void dfs(int index, int depth) {
if (Node[index].child.size() == 0) {
ans += Node[index].data * pow(1 + r, depth);
return;
}
for (int i = 0; i < Node[index].child.size(); i++) {
dfs(Node[index].child[i], depth + 1);
}
}
int main() {
int k, child;
(void)scanf("%d %lf %lf", &n, &p, &r);
r /= 100;
for (int i = 0; i < n; i++) {
(void)scanf("%d", &k);
if (k == 0) {
(void)scanf("%lf", &Node[i].data);//叶节点的存货量
}
else {
for (int j = 0; j < k; j++) {
(void)scanf("%d", &child);
Node[i].child.push_back(child);
}
}
}
dfs(0, 0);
printf("%.1f\n", p * ans);
return 0;
}