算法设计——分治法

问题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.

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值