1. 题目描述
给定一颗二叉树,树的每个节点的值为一个正整数。如果从根节点到节点 N 的路径上不存在比节点 N 的值大的节点,那么节点 N 被认为是树上的关键节点。求树上所有的关键节点的个数。请写出程序,并解释解题思路。
输入:3, 1, 4, 3, null, 1, 5
输出:4(图中蓝色节点是关键节点)
2. 我的尝试
这道题麻烦的地方在于输入与建树。由题意可知,输入数据是以完全二叉树形式进行输入的,因此可以考虑直接用数组来存储树,然后再遍历每一个节点并判断其是否为关键节点。
数据的输入是以空格作为分隔符的,因此不适宜直接用 cin
进行读入。可以用 getline
函数读入整行,再对读入的字符串遍历进行处理。
#include <bits/stdc++.h>
using namespace std;
int main() {
string str; // 存储输入的字符串
int val = 0; // 用于计算当前输入节点的值
vector<int> tree;
int cnt = 0;
getline(cin, str);
int n = str.size();
// 建树
for (int i = 0; i < n; i ++) {
auto c = str[i];
if ('0' <= c && c <= '9') {
val = val * 10 + c - '0';
} else if (val) {
tree.push_back(val);
val = 0;
} else if (c == 'n') {
tree.push_back(-1);
}
}
tree.push_back(val);
n = tree.size();
// 对除根节点外所有节点遍历,将其与各祖先节点比较,判断是否为关键节点
for (int i = 0; i < n && tree[i] != -1; i ++) {
int val = tree[i];
bool flag = true;
for (int p = (i - 1) / 2; p >= 0; p = (p - 1) / 2) {
if (tree[p] > val) flag = false;
if (p == 0) break;
}
if (flag) cnt ++;
}
// 根节点一定为关键节点,直接加1
cnt ++;
cout << cnt;
}