PAT（甲级）1079 Total Sales of Supply Chain (25point(s))

题目

题目链接

思路

题目大意：所有的货物都出自一个供货者，他出售的价格为p，然后卖给若干个经销商，经销商可以再次卖给经销商，价格提高进价的r%，最后一直到零售商的手里，零售商再卖给消费者；
为每个中间商定义一个结构体，如果child 为零，说明是零售商，此时需要计算他把手里所有商品卖出后的总收益；
如何计算每一个零售商的总收益，因为每经过一个中间商价格会增加 r%，所以可以算出零售商的深度（即叶子结点所在的深度，意味着转了几个中间商），最终收益 = 货物总量 * 供应商价格 * (1 + r%)^(叶子结点深度)

代码

#include <iostream>
#include <vector>
#include <stack>
#include <queue>
#include <algorithm>
#include <cstring>
#include <string>
#include <math.h>
using namespace std;
const int maxn = 1e5 + 5;
struct node{
     int data;
     vector<int> child;
}Node[maxn];

int n;//总人数
double p, r, ans = 0;//初始价格，增长率,总的价格
void DFS(int id, int depth){
     //如果没有孩子说明是叶节点，需要计算一下价格
     if(Node[id].child.size() == 0){
          ans += (Node[id].data * p * pow(1 + r, depth));
          return;
     }
     for(int i = 0; i < Node[id].child.size(); i ++){
          DFS(Node[id].child[i], depth + 1);
     }
     return;
}
int main()
{
     scanf("%d%lf%lf", &n, &p, &r);
     r /= 100;
     int num, t;
     for(int i = 0; i < n; i ++){
          scanf("%d", &num);
          if(num == 0){
               scanf("%d", &t);
               Node[i].data = t;
          }
          else{
               for(int j = 0; j < num; j ++){
                    scanf("%d", &t);
                    Node[i].child.push_back(t);
               }
          }
     }
     DFS(0, 0);
     printf("%.1f", ans);
     system("pause");
     return 0;
}