LeetCode二叉树小题目

Q1将有序数组转换为二叉搜索树

LeetCode二叉树小题目_第1张图片

题目大致意思就是从一个数组建立平衡的二叉搜索树。由于数组以及进行了升序处理,我们只要考虑好怎么做到平衡的。平衡意味着左右子树的高度差不能大于1。由此我们可以想着是否能用类似二分+递归来解决。

  • 如果left>right,直接返回nullpter

  • 否则 mid = (left + right) / 2,将a[mid]值赋给root结点

  • 递归左子树 right = mid-1;

  • 递归右子树left = mid+1;

这样一来问题就得到了解决。

class Solution {
public:

    TreeNode* sortedArrayToBST(vector& nums) {
        return build(nums,0,nums.size() - 1);
    }
    TreeNode*build(vector&nums,int left,int right)
    {
        if(left > right)
            return nullptr;
        int mid = (left + right) / 2;
        TreeNode*root = new TreeNode(nums[mid]);
        root->left = build(nums,left,mid-1);
        root->right = build(nums,mid+1,right);
        return root; 
    }
};

Q2路径总和

LeetCode二叉树小题目_第2张图片 本题是经典的搜索回溯+路径记录的题目。

路径记录:

  • 先序进行递归遍历,进行路径更新

  • 如果当前sum = root->val,将次路径添加到答案中

搜索回溯:

  • 递推参数: 当前节点 root ,当前目标值 tar

  • 终止条件: 若节点 root 为空,则直接返回。

  • 递推工作:

  • 路径更新: 将当前节点值 root.val 加入路径 path 。

  • 目标值更新: tar = tar - root.val(即目标值 tar 从 targetSum 减至 0 )

  • 路径记录: 当 (1) root 为叶节点 且 (2) 路径和等于目标值 ,则将此路径 path 加入 res 。

  • 先序遍历: 递归左 / 右子节点。

  • 路径恢复: 向上回溯前,需要将当前节点从路径 path 中删除,即执行 path.pop_back() 。

    class Solution {
    public:
      vector> ret;
      vector path;
      void dfs(TreeNode*root,int targetSum)
      {
          if(root==nullptr)return;
          path.push_back(root->val);
          if(root->left == nullptr && root->right == nullptr)
          {
              if(targetSum == root->val)
                  ret.push_back(path);
          }
          dfs(root->left,targetSum-root->val);
          dfs(root->right,targetSum-root->val);
          path.pop_back();
      }
      vector> pathSum(TreeNode* root, int targetSum) {
         dfs(root,targetSum);
         return ret;
      }
    };

    本文由博客一文多发平台 OpenWrite 发布!

你可能感兴趣的:(青少年编程,c++)