【华为OD机试真题】查找树中元素（C++）[详细思路&非专栏]

题目描述：

题目源引

已知树形结构的所有节点信息，现要求根据输入坐标（x,y）找到该节点保存的内容值，其中x表示节点所在的层数，根节点位于第0层，根节点的子节点位于第1层，依次类推；y表示节点在该层内的相对偏移，从左至右，第一个节点偏移0，第二个节点偏移1，依次类推。

上图中，假定圆圈内的数字表示节点保存的内容值，则根据坐标(1,1)查到的内容值是23。

每个节点以一维数组（int[]）表示，所有节点信息构成二维数组（int[][]），二维数组的0位置存放根节点；

表示单节点的一维数组中，0位置保存内容值，后续位置保存子节点在二维数组中的索引位置。

对于上图中：

1. 根节点的可以表示为{10，1，2}
2. 树的整体表示为{{10,1,2},{-21,3,4},{23,5},{14},{35},{66}}

---

输入

1. 第一行：一个整数n，表示节点数量
2. 之后的n行：每行第一个数表示当前节点的值，其余的值表示当前节点的子节点索引，用空格隔开
3. 最后一行：两个整数，用空格隔开，表示待检索的节点的索引

---

输出：

• 如果待检索节点有值，则返回其值。（形如：{23}）
• 如果待检索节点没有值，则返回空。（形如：{}）

---

示例：

输入：
6
10 1 2
-21 3 4
23 5
14
35
66
1 1

输出：
{23}

输入：
14
0 1 2 3 4
-11 5 6 7 8
113 9 10 11
24 12
35
66 13
77
88
99
101
102
103
25
104
3 2
输出：
{}

---

解题

本题属于树的层次遍历。楼主提供算法(C++)如下：

#include <iostream>
#include <vector>
#include <queue>

using namespace std;

int main()
{
    // 1.初始化
    int nodes; // 节点数量
    cin >> nodes;
    vector<vector<int>> arr(nodes, vector<int>()); // 保存键入节点信息的二维数组
    for (int i = 0; i < nodes; i++)
    {
        int val;
        do
        {
            cin >> val;
            arr[i].push_back(val);
        } while (getchar() != '\n');
    }

    int rows, cols; // 待检索节点的索引位置
    cin >> rows >> cols;

    // 2.构建树的层次遍历二维数组
    vector<vector<int>> mat; // 保存整棵树层次遍历（从左到右）的节点值
    queue<int> que;          // 暂存层次遍历节点索引的队列

    vector<int> root;
    root.push_back(arr[0][0]);
    mat.push_back(root);                    // 插入头节点的值
    for (int i = 1; i < arr[0].size(); i++) // 将头节点的子节点索引插入队列
    {
        que.push(arr[0][i]);
    }

    while (!que.empty())
    {
        vector<int> vec;       // 用于保存当前层节点的值
        int nums = que.size(); // 当前层节点的个数
        for (int i = 0; i < nums; i++)
        {
            int node_idx = que.front(); // 当前节点索引
            que.pop();

            vec.push_back(arr[node_idx][0]); // 保存当前节点的值
            for (int j = 1; j < arr[node_idx].size(); j++)
            {
                que.push(arr[node_idx][j]); // 将当前节点的子节点索引插入队列
            }
        }
        mat.push_back(vec);
    }

    // 3.根据输入的索引检索
    if (rows < mat.size() && cols < mat[rows].size())
    {
        cout << "{" << mat[rows][cols] << "}" << endl;
    }
    else
    {
        cout << "{}" << endl;
    }

    return 0;
}