力扣剑指offer第17天树

最新推荐文章于 2024-10-16 10:13:45 发布

slam_SponseBob

最新推荐文章于 2024-10-16 10:13:45 发布

阅读量170

点赞数

文章标签： leetcode 链表算法

本文链接：https://blog.csdn.net/weixin_51650301/article/details/122776978

版权

本文档介绍了三种与二叉树相关的问题解决方案：1) 计算向下的路径节点之和，使用深度优先搜索；2) 寻找节点之和最大的路径，采用深度优先遍历求解；3) 展平二叉搜索树，通过中序遍历实现。这些技术展示了在算法设计中对树形数据结构的处理技巧。

摘要由CSDN通过智能技术生成

50）向下的路径节点之和

class Solution{
public:
    int pathSum(TreeNode* root, int targetSum){
        if(!root) return 0;
        int ret = rootSum(root, targetSum);
        ret += pathSum(root->left, targetSum);
        ret += pathSum(root->right, targetSum);
        return ret;
    }

    int rootSum(TreeNode* root, int sum){
        if(!root) return 0;
        int ret = 0;
        if(sum==root->val) ret++;
        ret += rootSum(root->left, sum-root->val);
        ret += rootSum(root->right, sum-root->val);
        return ret;
    }
};

51)节点之和最大的路径

class Solution{
private:
    int ret;
    int dfs(TreeNode* root){
        if(!root) return 0;
        int left = max(0, dfs(root->left));
        int right = max(0, dfs(root->right));
        ret = max(ret, left+right+root->val);
        return root->val + max(left, right);
    }
public:
    int maxPathSum(TreeNode* root){
        ret = INT_MIN;
        dfs(root);
        return ret;
    }
};

52)展平二叉搜索树

class Solution {
public:
    void inorder(TreeNode* root, vector<int>& res){
      if(!root) return;
      inorder(root->left, res);
      res.push_back(root->val);
      inorder(root->right, res);
    }
    
    TreeNode* increasingBST(TreeNode* root) {
      vector<int> res;
      inorder(root, res);
      TreeNode* dummynode = new TreeNode(-1);
      TreeNode* curnode = dummynode;
      for(int item : res){
        curnode->right = new TreeNode(item);
        curnode = curnode->right;
      }
      return dummynode->right;
    }
};