二叉树中所有路径,以及满足求和为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;
}
};