问题1.最大二叉树
给定一个不含重复元素的整数数组 numsnums。以此数组直接递归构建的最大二叉树。最大二叉树定义如下:
- 二叉树的根是数组 numsnums 中的最大元素。
- 左子树是通过数组中最大值左边部分递归构造出的最大二叉树。
- 右子树是通过数组中最大值右边部分递归构造出的最大二叉树。
返回有给定数组 numsnums 构建的最大二叉树 。
输入
输入一行多个数字代表数组 numsnums。数字与数字之间用空格隔开。
以 EOF
作为结束输入(请参考 SCNUOJ 1001 题题解了解如何处理输入)。
- 1≤nums.length≤10001≤nums.length≤1000
- 0≤nums[i]≤10000≤nums[i]≤1000
- numsnums 中的所有整数互不相同
输出
输出一行字符代表二叉树的构造结果(前序遍历),具体请参考样例
样例
标准输入复制文本 |
3 2 1 6 0 5 |
标准输出复制文本 |
6 3 null 2 null 1 5 0 null |
#include<iostream>//输入输出,cout和cin
#include<vector>//动态数组,使用push_back()和size()
#include<algorithm>//算法库,std::sort对元素进行排序等等
using namespace std;
struct TreeNode {//构建二叉树
public:
int val;//存储树结点的值
TreeNode* left;//指向左子树的指针,左子节点
TreeNode* right;//指向右子树的指针,右子节点
TreeNode() {}
TreeNode(int val) : val(val), left(nullptr), right(nullptr) {}//二叉树初始化,左子树指针初始化位nullptr;右子树指针初始化nullptr。表明这个二叉树目前为空
};
TreeNode* buildTree(vector<int> vt) {//定义了一个函数名为buildTree的函数,该函数接受一个整数向量vt作为参数,并返回一个指向TreeNode结构体的指针
int maxValue = *max_element(vt.begin(), vt.end());//vt.begin()返回指向向量vt第一个元素的迭代器,vt.end()返回指向向量vt的最后一个元素的迭代器。而max_element()是算法库的一个算法,用于找到vt这个序列中的最大元素。然后*运算符是用于解引用迭代器,以获取迭代器指向的值,也就是vt的最大值。然后将这个最大值存储在变量maxValue中。
int maxIndex = 0;
for (auto v : vt) {
if (v == maxValue) {
break;
}
maxIndex++;
}
TreeNode* root = new TreeNode(maxValue);
vector<int> L(vt.begin(), vt.begin() + maxIndex), R(vt.begin() + maxIndex + 1, vt.end());
/*
vector<int> L, R;
for (int i = 0; i < vt.size(); ++i) {
if (i < maxIndex) {
L.push_back(vt[i]);
}
else if (i > maxIndex) {
R.push_back(vt[i]);
}
}
*/
if (!L.empty()) root->left = buildTree(L);
if (!R.empty()) root->right = buildTree(R);
return root;
}
void prev_traversal(TreeNode* root, vector<int>& traversal) {
if (!root) {
traversal.push_back(-1);
return;
}
traversal.push_back(root->val);
if (root->left || root->right) {
prev_traversal(root->left, traversal);
prev_traversal(root->right, traversal);
}
}
int main() {
int n;
vector<int> vt;
while (cin >> n){
vt.push_back(n);
}
TreeNode* root = buildTree(vt);
vector<int> traversal;
prev_traversal(root, traversal);
for (int i = 0; i < traversal.size(); ++i) {
if (!i) {
cout << traversal[i];
}
else if (traversal[i] == -1) {
cout << " null";
}
else {
cout<< " " << traversal[i];
}
}
return 0;
}
结题思路以及代码分析:
1.