【PAT甲级 树/图的遍历】1079 Total Sales of Supply Chain (25 分)

BFS和DFS都写了

# include <bits/stdc++.h>
using namespace std;

int N;
double P, r; 

vector< vector<int> > G;
map<int, int> leafProductAmount; // 这里面的键都是叶子结点，值都是叶子结点的product 的数量
vector<int> level;

// 广度优先遍历求路径长度（或者树的层数）
void BFS(){
    queue<int> Q;
    Q.push(0);
    level[0] = 0;
    while(!Q.empty()){
        int u = Q.front();
        Q.pop();
        for(int v: G[u]){
            level[v] = level[u] + 1;
            Q.push(v);
        }
    }
}

void DFS(int u, int lv){
    // 叶子结点
    if(G[u].size() == 0){
        level[u] = lv; // 只算叶子节点的level
        return;
    }
    // level[u] = lv; // 计算非叶子结点的level（可不写）
    // 前往子节点
    for(int v: G[u]){
        DFS(v, lv+1);
    }
}

int main(){
    cin >> N >> P >> r;
    level = vector<int>(N);
    G.resize(N);
    for(int i = 0;i < N;++i){
        int K;
        cin >> K;
        if(K != 0){
            for(int j = 0;j < K;++j){
                int v;
                cin >> v;
                G[i].push_back(v);
            }
        } else {
            cin >> leafProductAmount[i];
        }
    }
//     BFS();
    DFS(0, 0);
    double rst = 0.0;
    for(auto p: leafProductAmount){
        rst += P * pow((1 + 0.01*r), level[p.first]) * p.second;
    }
    printf("%.1lf\n", rst);
    
    return 0;
}