【C++】二叉树中路径系列问题

二叉树中所有路径,以及满足求和为N的子路径等系列问题求解

0X00. 问题1:二叉树根到叶子节点所有路径总数

给你一个二叉树的根节点 root ,按任意顺序 ,返回所有从根节点到叶子节点的路径。叶子节点 是指没有子节点的节点。

class Solution {
public:
    vector<int> tmp;
    vector<string> res;
    vector<string> binaryTreePaths(TreeNode* root) {
        if (root==nullptr) return res;
        dfs(root); 
        return res;

    }
    void dfs(TreeNode* root) {
        if (root==nullptr) return;
        tmp.push_back(root->val);
        // 1. path += to_string(root->val);
        if (root->left==nullptr&&root->right==nullptr) {
            res.push_back(transStr(tmp));
        }
        binaryTreePaths(root->left);
        // 2. binaryTreePaths(root->left, path+"->");
        binaryTreePaths(root->right);
        tmp.pop_back();
    }
    // 通俗易懂的写法,为了降低空间时间复杂度,可以将以下步骤放到迭代过程中进行
    string transStr(vector<int> num) {
        string str;
        for (int i=0; i<num.size(); i++) {
            
            if (i==num.size()-1) {
                str += to_string(num[i]);
                break;
            }
            str += to_string(num[i])+"->";
        }
        return str;
    }
};
0X01. 问题2:是否有根到叶子节点总和为N的路径

给你二叉树的根节点 root 和一个表示目标和的整数 targetSum 。判断该树中是否存在 根节点到叶子节点 的路径,这条路径上所有节点值相加等于目标和 targetSum 。如果存在,返回 true ;否则,返回 false 。
叶子节点 是指没有子节点的节点。

class Solution {
public:
    bool hasPathSum(TreeNode* root, int targetSum) {
        if (root== nullptr) return false;
        if (root->left==nullptr && root->right==nullptr) // 这一句是判断为叶子节点,如果根节点本来就没子节点那么这个时候root直接判断是否满足条件
            return targetSum-root->val == 0;
        return hasPathSum(root->left, targetSum-root->val) || hasPathSum(root->right, targetSum-root->val);
    }
};
0X02. 问题3:根到叶子节点总和为N的所有路径

给你二叉树的根节点 root 和一个整数目标和 targetSum ,找出所有 从根节点到叶子节点 路径总和等于给定目标和的路径。
叶子节点 是指没有子节点的节点。

class Solution {
public:
    vector<vector<int>> pathSum(TreeNode* root, int targetSum) {
        vector<vector<int>> res;
        if (root==nullptr) return res;
        vector<int> tmp;
        dfs(res, tmp, root, targetSum);
        return res;
    }
    void dfs( vector<vector<int>> &res, vector<int> &tmp, TreeNode* root, int sum) {

        if (root==nullptr) return;
        tmp.push_back(root->val);
        if (root->left==nullptr && root->right==nullptr && sum==root->val) {
            res.push_back(tmp);
        }
            
        dfs(res, tmp, root->left, sum-root->val);
        dfs(res, tmp, root->right, sum-root->val);
        tmp.pop_back();
    }
};
0X03. 问题4:总和为N的所有路径(不一定是从根节点到子节点这一段)

给定一个二叉树的根节点 root ,和一个整数 targetSum ,求该二叉树里节点值之和等于 targetSum 的 路径 的数目。
路径 不需要从根节点开始,也不需要在叶子节点结束,但是路径方向必须是向下的(只能从父节点到子节点)。

class Solution {
public:
    int path=0;
    int pathSum(TreeNode* root, int targetSum) {
        if (root==nullptr) return 0;
		// 双层递归,从每个节点出发都有可能
        dfs(root, targetSum);

        pathSum(root->left, targetSum);
        pathSum(root->right, targetSum);
        return path;
    }
    void dfs(TreeNode* root, int sum) {
        if (!root) return;
        if (sum-root->val == 0) path++; // 只要遍历过程中满足了条件就++
        dfs(root->left, sum-root->val);
        dfs(root->right, sum-root->val);
    }
};
0X04. 问题5:二叉树中最大路径和

路径 被定义为一条从树中任意节点出发,沿父节点-子节点连接,达到任意节点的序列。同一个节点在一条路径序列中 至多出现一次 。该路径 至少包含一个 节点,且不一定经过根节点。
路径和 是路径中各节点值的总和。

class Solution {
public:
	int res=INT_MIN;
	int maxPathSum(TreeNode* root) {
		dfs(root);
		return res;
	}
	int dfs(TreeNode* root) {
		if (root==nullptr) return 0;
		int left = max(0, dfs(root->left));
		int right = max(0, dfs(root->right));
		res = max(res, root->val+left+right);
		return max(left, right) + root->val;
	}
 };
0X05. 问题6:二叉树的直径
class Solution {
public:
	int res = 0;
	int maxPath(TreeNode* root) { 
		dfs(root);
		return res;
	}
	int dfs (TreeNode* root) {
		if (root==nullptr) return;
		int left = dfs(root->left);
		int right = dfs(root->right);
		res = max(left+right, res);
		return max(left, right)+1;
	}
};
0X06. 问题7:687. 最长同值路径

给定一个二叉树的 root ,返回 最长的路径的长度 ,这个路径中的 每个节点具有相同值 。 这条路径可以经过也可以不经过根节点。
两个节点之间的路径长度 由它们之间的边数表示。
例子:
输入:root = [5,4,5,1,1,5]
输出:2

这道题本质上与上边的二叉树的直径相似,主要是会在递归过程中加上判断条件,判断root->left或root->right和root的值是否相等,然后再判断子树的深度。

class Solution {
public:
    int maxSize=0; 
    int longestUnivaluePath(TreeNode* root) {
        if (root==nullptr) return 0;
        dfs(root);
        return maxSize;
    }
    int dfs(TreeNode* root) {
        if (root==nullptr) return 0;
        int leftsize = dfs(root->left);
        int rightsize = dfs(root->right);
        if (leftsize>0 && root->left->val!=root->val) leftsize=0;
        if (rightsize>0 && root->right->val!=root->val) rightsize=0;
        maxSize = max(maxSize, leftsize+rightsize);
        return max(leftsize, rightsize)+1;
    }
};
  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
可以使用递归的方式来实现求解二叉树所有路径的功能。具体实现方法如下: 1. 定义一个函数,输入参数为当前节点指针、当前路径字符串和结果数组。 2. 如果当前节点为空,则直接返回。 3. 将当前节点的值添加到当前路径字符串。 4. 如果当前节点为叶子节点,则将当前路径字符串添加到结果数组。 5. 否则,递归遍历当前节点的左子树和右子树,分别调用上述函数。 6. 递归结束后,将当前节点的值从当前路径字符串删除,以便遍历其它路径。 下面是具体的 C 代码实现: ``` #include <stdio.h> #include <stdlib.h> #include <string.h> #define MAX_PATH_LEN 100 struct TreeNode { int val; struct TreeNode *left; struct TreeNode *right; }; void binaryTreePathsHelper(struct TreeNode* node, char* path, char** result, int* returnSize) { if (node == NULL) { return; } int len = strlen(path); if (len > ) { path[len] = '-'; path[len+1] = '>'; path[len+2] = '\'; } char val_str[10]; sprintf(val_str, "%d", node->val); strcat(path, val_str); if (node->left == NULL && node->right == NULL) { result[*returnSize] = (char*)malloc(sizeof(char) * (strlen(path) + 1)); strcpy(result[*returnSize], path); (*returnSize)++; } else { binaryTreePathsHelper(node->left, path, result, returnSize); binaryTreePathsHelper(node->right, path, result, returnSize); } path[strlen(path) - strlen(val_str) - 2] = '\'; } char ** binaryTreePaths(struct TreeNode* root, int* returnSize) { char** result = (char**)malloc(sizeof(char*) * MAX_PATH_LEN); *returnSize = ; char path[MAX_PATH_LEN]; path[] = '\'; binaryTreePathsHelper(root, path, result, returnSize); return result; } int main() { struct TreeNode* root = (struct TreeNode*)malloc(sizeof(struct TreeNode)); root->val = 1; root->left = (struct TreeNode*)malloc(sizeof(struct TreeNode)); root->left->val = 2; root->left->left = NULL; root->left->right = (struct TreeNode*)malloc(sizeof(struct TreeNode)); root->left->right->val = 5; root->left->right->left = NULL; root->left->right->right = NULL; root->right = (struct TreeNode*)malloc(sizeof(struct TreeNode)); root->right->val = 3; root->right->left = NULL; root->right->right = NULL; int returnSize; char** result = binaryTreePaths(root, &returnSize); for (int i = ; i < returnSize; i++) { printf("%s\n", result[i]); } return ; } ``` 运行结果如下: ``` 1->2->5 1->3 ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

郝同学

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值