本地由LeetCode输入构建二叉树(C++版)

引言

LeetCode里涉及到二叉树的题都是采用了形如[1,2,3,4,NULL,5,NULL]这样的格式作为输入,其中NULL就表示该节点为空节点。LeetCode并没有给出如何通过这样一个数组进行二叉树的构建,但我们通过分析可以发现这实际上就是一个对二叉树的层次遍历过程,只不过空的叶子节点使用了NULL来进行占位。

方案

为了方便在本地进行Debug,我们可以依据这一流程自己构造对应的二叉树。考虑到这一格式输入本身就是某种意义上的层次遍历,因此我们也只需要以层次遍历的方式来建树即可。层次遍历的写法中,我们是初始化了一个队列,循环地在每一层执行从左到右输出,从左到右子节点入队列的操作,代码为:

cpp 复制代码
 vector<vector<int>> result;
 queue<TreeNode*> que;
 if(root) que.push(root);
 while(!que.empty()){
     vector<int> cur;
     int size=que.size();
     for(;size>0;size--){
         TreeNode* temp=que.front();
         que.pop();
         cur.push_back(temp->val);
         if(temp->left) que.push(temp->left);
         if(temp->right) que.push(temp->right);
     }
     result.push_back(cur);
 }
 return result;

同理,我们也利用队列以层次遍历的方式来实现二叉树的构造。同样是遍历每一层,依据前一层的大小size来决定本层操作要从输入vector中读取多少个元素来构造子节点,每次构造的顺序为从左至右。考虑到输入使用了NULL来进行占位,使用迭代器遍历时读出的(*it)==0 ,因此我们要对这个空节点进行额外判断,如果为0,则当前处理节点的子节点为空,否则new一个值为(*it)的子节点。其余逻辑比较简单,不再赘述。具体代码见下:

cpp 复制代码
class TreeNode{
    public:
        int val;
        TreeNode* left;
        TreeNode* right;
        TreeNode():val(0),left(NULL),right(NULL){}
        TreeNode(int v):val(v),left(NULL),right(NULL){}
};
cpp 复制代码
TreeNode* CreatTree(vector<int> input){
    TreeNode* root;
    queue<TreeNode*> que;
    vector<int>::iterator it=input.begin();
    if(!input.empty()){
        root = new TreeNode(*(it++));
        que.push(root);
    }
    while(!que.empty() && it!=input.end()){
        int size=que.size();
        for(;size--;size>0){
            TreeNode* temp=que.front();
            que.pop();
            TreeNode* left=nullptr;
            TreeNode* right=nullptr;
            if (*it) {
                left = new TreeNode(*(it++));
            } else {
                it++;
            }

            if (*it) {
                right = new TreeNode(*(it++));
            } else {
                it++;
            }
            temp->left=left;
            temp->right=right;
            if(left) que.push(left);
            if(right) que.push(right);
        }
        
    }
    return root;
}
相关推荐
??tobenewyorker17 分钟前
力扣打卡第二十一天 中后遍历+中前遍历 构造二叉树
数据结构·c++·算法·leetcode
蓝澈112125 分钟前
迪杰斯特拉算法之解决单源最短路径问题
java·数据结构
oioihoii1 小时前
C++11 forward_list 从基础到精通:原理、实践与性能优化
c++·性能优化·list
m0_687399841 小时前
写一个Ununtu C++ 程序,调用ffmpeg API, 来判断一个数字电影的视频文件mxf 是不是Jpeg2000?
开发语言·c++·ffmpeg
呆瑜nuage3 小时前
数据结构——堆
数据结构
Ronin3053 小时前
【C++】类型转换
开发语言·c++
蓝澈11213 小时前
弗洛伊德(Floyd)算法-各个顶点之间的最短路径问题
java·数据结构·动态规划
zl_dfq3 小时前
数据结构 之 【堆】(堆的概念及结构、大根堆的实现、向上调整法、向下调整法)(C语言实现)
数据结构
127_127_1273 小时前
2025 FJCPC 复建 VP
数据结构·图论·模拟·ad-hoc·分治·转化
闪电麦坤953 小时前
数据结构:二维数组(2D Arrays)
数据结构·算法