华为机考真题 -- 传递悄悄话

题目描述：

给定一个二叉树，每个节点上站着一个人，节点数字表示父节点到该节点传递悄悄话需要花费的时间。初始时，根节点所在位置的人有一个悄悄话想要传递给其他人，求二叉树所有节点上的人都接收到悄悄话花费的时间。

输入描述：给定二叉树 0 9 20 -1 -1 15 7 -1 -1 -1 -1 3 2
注：-1 表示空节点

输出描述：返回所有节点都接收到悄悄话花费的时间 38

示例1：
输入
0 9 20 -1 -1 15 7 -1 -1 -1 -1 3 2

输出
38

C++源码：

#include <iostream>
#include <deque>
#include <vector>
#include <sstream>
#include <string>

using namespace std;

void checkAllNode(int index, vector<int>& nums, int fNode, deque<int>& queue, int& allTime) {
    if (index < nums.size() && nums[index] != -1) {
        nums[index] += nums[fNode];
        queue.push_back(index);
        if (nums[index] > allTime) {
            allTime = nums[index];
        }
    }
}

int main() {
    string str;
    getline(cin, str);
    vector<int> nums;
    stringstream ss(str);
    int num;
    while (ss >> num) {
        nums.push_back(num);
        ss.ignore(1); // ignore the space
    }

    deque<int> queue;
    int allTime = 0;
    queue.push_back(0);
    while (!queue.empty()) {
        int fNode = queue.front();
        queue.pop_front();
        int temp = 2 * fNode;
        checkAllNode(temp + 1, nums, fNode, queue, allTime);
        checkAllNode(temp + 2, nums, fNode, queue, allTime);
    }

    cout << allTime << endl;

    system("pause");
    return 0;
}