Leetcode中使用数组创建二叉树

Leetcode中树的题目大多是输入数组的，而在debug的时候还是会有点问题的，所以就自己写了一个通过数组创建二叉树的代码，如下：

#include <iostream>
#include <vector>

using namespace std;

#define null INT_MIN / 2 // null 可自己按题目定义

/// Leetcode中树节点的结构体
struct TreeNode {
  int val;
  TreeNode *left;
  TreeNode *right;
  TreeNode() : val(0), left(nullptr), right(nullptr) {}
  TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
  TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
};


TreeNode* BuildTree(const vector<int>& vec, int index/*当前节点在数组中的下标*/) {
  TreeNode* root = nullptr;

  if (index < vec.size() && vec[index] != null) {
    root = new TreeNode(vec[index]);
    root->left = BuildTree(vec, 2 * index + 1);
    root->right = BuildTree(vec, 2 * index + 2);
  }

  return root;
}

/// 先序遍历，用来验证创建的树对不对
void preorder(TreeNode* root) {
  if (root == nullptr) return;
  cout << root->val << " ";
  preorder(root->left);
  preorder(root->right);
}

OK, 用来验证以下，这里以98. 验证二叉搜索树中的示例 2来测试，

int main() {
  // 例子中的为[5,1,4,null,null,3,6], 这里用来初始化vector
  vector<int> vec = { 5,1,4,null,null,3,6 }; 
  auto root = BuildTree(vec, 0);
  preorder(root);

  return 0;
}

看看输出的是不是5 1 4 3 6，如果是则和预期一样。