二叉树高频题目——下——不含树型dp
一,普通二叉树上寻找两个节点的最近的公共祖先
1,介绍
LCA(Lowest Common Ancestor,最近公共祖先)是二叉树中经常讨论的一个问题。给定二叉树中的两个节点,它的LCA是指这两个节点的最低(最深)的公共祖先节点。这个问题常见于计算机科学和算法设计中,具体的问题可以是:
-
二叉搜索树(BST)中的LCA问题:给定两个节点,找出它们在BST中的最近公共祖先。
-
普通二叉树中的LCA问题:在普通的二叉树中,不一定是BST,也可以找出任意两个节点的最近公共祖先。
这个问题在树的数据结构和算法中很常见,通常可以通过深度优先搜索(DFS)或者广度优先搜索(BFS)等方法来解决。
深度优先搜索(DFS)和广度优先搜索(BFS)是两种常见的图和树的遍历算法,它们在解决许多问题中都非常有用。这里分别介绍它们的特点和应用场景:
深度优先搜索(DFS)
特点:
- 深度优先搜索是一种通过从根节点开始沿着树的深度遍历的算法。
- 在实现上,通常使用递归或显式的栈来存储遍历过程中的节点。
- 深度优先搜索会一直向深度方向探索,直到达到树的底部,然后再回溯到前面的节点进行其他路径的探索。
应用场景:
- 查找路径:在图或树中查找从一个节点到另一个节点的路径。
- 拓扑排序:在有向无环图(DAG)中对节点进行排序,使得所有的有向边从排在前面的节点指向排在后面的节点。
- 连通性检测:判断两个节点之间是否有路径相连。
- 生成迷宫:利用深度优先搜索来生成迷宫的布局。
广度优先搜索(BFS)
特点:
- 广度优先搜索是一种通过逐层遍历的方式,从根节点开始向外扩展搜索的算法。
- 使用队列来存储当前层级的所有节点,以确保按层级顺序处理。
- 从根节点开始,首先遍历所有与根节点直接相连的节点,然后依次遍历与这些节点相连的节点,以此类推。
应用场景:
- 最短路径:在无权图中查找从一个节点到另一个节点的最短路径。
- 最小生成树:在加权图中找到连接所有节点的最小权重的边的集合。
- 网络广播:在社交网络或信息传播中找到最快的传播路径。
- 解决迷宫问题:利用广度优先搜索来寻找迷宫中的最短路径或所有可能的路径。
总结
深度优先搜索和广度优先搜索在解决不同类型的问题时具有不同的优势和适用性。选择哪种算法取决于具体的问题和要求,例如是否需要最短路径、是否有环等因素。这两种算法都是基础且重要的数据结构和算法设计中的核心内容。
2,分析
假设有个二叉树:
1
/ \
2 3
/ \ / \
4 5 6 7
/ \
8 9
共同祖先有两种情况:
1)包含关系,例如4和8,共同祖先就是4.
2)并列关系,例如8和9,共同祖先就是4.
3,代码
Node* lowestCommonAncestor(Node* root, Node* p, Node* q)
{
if (root == NULL || root == p || root == q) return root;
//遇到空、q、p节点时直接返回,不再额外声明变量,直接把root当作临时变量。
//左树搜索p和q,同样遇到p、q、空节点时直接返回。
Node* l = lowestCommonAncestor(root->left, p, q);
//右树搜索p和q,同样遇到p、q、空节点时直接返回。
Node* r = lowestCommonAncestor(root->right, p, q);
if (l != NULL && r != NULL) return root;
//左树搜索到,右树也搜索到,返回root。
if (l == NULL && r == NULL) return NULL;
//如果都没有搜索到的话,就返回空。
//l和r中有一个为空,一个不为空,就返回不为空的那一个。
return l != NULL ? l : r;
}
4,分析
难以理解,当然,我第一次看的时候也是一脸懵逼。
接下来分别分析:
第一种情况,就举2和5.
进行函数递归调用时遇到2,就直接开始返回2,后面的子树部分根本不会遍历。而右边3这部分子树只会返回NULL,最后就返回2.
第二种情况,举4和5吧
二,在搜索二叉树上寻找两个节点的最近公共祖先
1,介绍——搜索二叉树
搜索二叉树(Binary Search Tree,BST)是一种二叉树的特殊形式,具有以下性质:
-
结构特点:
- 每个节点最多有两个子节点,分别称为左子节点和右子节点。
- 对于每个节点,其左子树中的所有节点的值都小于该节点的值。
- 对于每个节点,其右子树中的所有节点的值都大于该节点的值。
- 左右子树本身也是BST。
-
排序性质:
- 中序遍历BST可以按照升序输出节点的值。
- 这是因为对于任意节点,其左子树中的值都小于该节点,右子树中的值都大于该节点,因此中序遍历会按顺序访问节点。
-
操作:
- 插入:将新值插入到BST中正确的位置,保持BST的性质。
- 搜索:根据给定值搜索节点。
- 删除:删除特定节点,并保持BST的性质。
搜索二叉树在数据结构中非常常见,因为它具有高效的搜索、插入和删除操作。然而,如果插入的节点顺序不当,可能导致BST失衡,影响其性能。
2,思路
假如找节点q和p,而 p 小于q。
1)当前节点小于p的话,就往右移到右节点,因为根据搜索二叉树的性质,左子树节点一定小于p。
2)当前节点大于q的话,就往左移到左节点。
3)当前节点大于p,小于q的话,直接返回该节点,一定是第一次遇到大于p小于q的节点。因为你想,如果第一次遇到后你还继续移动,后面节点的值都要大于第一次遇到节点的值,都要大于p的。
4)如果遇到p或q,直接返回。
3,补充
函数重载
4,代码
Node* lowestCommonAncestor_BST(Node* root, Node* p, Node* q)
{
int min = p->value < q->value ? p->value : q->value;
int max = p->value > q->value ? p->value : q->value;
//root从上到下。
while (root->value != p->value && root->value != q->value)
{
//while里的条件是没有遇到p或q就继续遍历。
if (min < root->value && root->value < max) break;
root = root->value < min ? root->right : root->left;
}
return root;
}
Node* lowestCommonAncestor_BST(Node* root, int p, int q)
{
int min = p < q ? p : q;
int max = p > q ? p : q;
//root从上到下。
while (root->value != p && root->value != q)
{
//while里的条件是没有遇到p或q就继续遍历。
if (min < root->value && root->value < max) break;
root = root->value < min ? root->right : root->left;
}
return root;
}
三,收集累加和等于aim的所有路径(递归恢复现场)
只收集从根节点一直走到叶节点的路径。
1,介绍
回溯(Backtracking)
在算法中,回溯(Backtracking)是一种递归的技术,常用于解决组合优化问题和搜索问题。它通过尝试所有可能的候选解,并在搜索过程中逐步构建解决方案。如果当前尝试的部分解不符合要求,回溯算法会放弃这个部分解,并尝试下一个候选解。
具体来说,回溯算法通常适用于如下情况:
-
决策树的遍历:通过构建一个决策树,每个节点代表问题的一部分解,从根节点开始逐步扩展,直到找到满足条件的解或者无法继续为止。
-
状态空间搜索:在搜索问题中,每个状态表示问题的一个可能状态,通过深度优先搜索(DFS)遍历状态空间,寻找解决方案。
回溯算法通常包括以下步骤:
- 选择:根据当前状态选择一个候选解。
- 约束:检查候选解是否满足问题的所有约束条件。
- 递归:如果候选解满足约束条件,则递归地继续尝试下一个部分解。
- 回溯:如果当前部分解无法继续扩展或者不符合条件,则回溯到上一步,尝试其他候选解。
回溯算法的经典应用包括八皇后问题、数独、组合求和等。它的优点在于简单易懂,并且能够找到所有解或者最优解(取决于具体实现)。然而,回溯算法的缺点是在大规模问题上可能会耗费大量时间,因为它会尝试所有可能的组合。
2,思路
将aim设为全局变量,同时准备一个全局变量来记录遍历过路径节点,在到达叶节点后发现满足条件就加入到大结果中去。
因为我们使用递归,要避免上一次的结果影响,所以在函数调用结束时抹除上一次的记录。
3,代码
假设我的aim是10,就有两个路径5->4->1和5->3->2。
错误
void process(Node* cur, int aim, int sum, list<int> path, list<list<int>> ans)
这样传参是错误的,因为函数要修改全局变量,这是值传递,应该改为引用传递。
正确
list<list<int>> pathSum(Node* root, int aim)
{
//先准备好全局变量
list<list<int>> ans;
if (root != NULL)
{
list<int> path;
process(root, aim, 0, path, ans);
}
return ans;
}
void process(Node* cur, int aim, int sum, list<int>& path, list<list<int>>& ans)
{
//注意sum不包括cur的值,而是cur上方经过节点值的总和。
if (cur->left == NULL && cur->right == NULL)
//当我遇到叶节点的时候
{
if (cur->value + sum == aim)
{
//当满足目标值后,将该节点的值加入到path里去
//同时复制一份到ans里面。
path.push_back(cur->value);
list<int> copylist(path);
ans.push_back(copylist);
path.pop_back();
}
}
else
//不是叶节点的时候
{
path.push_back(cur->value);
if (cur->left != NULL)
{
process(cur->left, aim, sum + cur->value, path, ans);
}
//去探索左边的路径有没有等于aim,如果有就会加入到ans。
if (cur->right != NULL)
{
process(cur->right, aim, sum + cur->value, path, ans);
}
//探索完后,弹出cur,避免影响递归调回后的遍历。在回溯时会弹出节点。
path.pop_back();
//取名:恢复现场
}
}
四,验证平衡二叉树(树型dp沾边)
1,平衡二叉树
平衡二叉树(Balanced Binary Tree)是一种特殊的二叉树,其特点是在插入和删除节点后,树的高度能够保持在一个相对较低的水平,从而确保各种操作(如查找、插入和删除)的时间复杂度保持在对数级别。平衡二叉树在计算机科学中广泛应用于各种数据结构和算法中,如数据库索引、内存管理以及各种需要高效查找的数据结构。
平衡二叉树的定义
平衡二叉树是一种二叉搜索树(Binary Search Tree,BST),它满足以下平衡条件之一:
-
高度平衡(Height-Balanced):
每个节点的左右子树高度差不超过1。这种平衡条件常见于AVL树。 -
黑平衡(Black-Height Balanced):
对于每个节点,从该节点到所有叶子节点的路径上,黑色节点的数量相同。这是红黑树的平衡条件。 -
其他平衡条件:
不同的平衡二叉树可能采用不同的平衡条件,如Splay树通过自适应调整保持平衡。
常见的平衡二叉树类型
1. AVL树(Adelson-Velsky and Landis Tree)
AVL树是最早提出的自平衡二叉搜索树之一,由G.M. Adelson-Velsky和E.M. Landis在1962年提出。其特点是:
- 平衡因子:每个节点的左子树和右子树的高度差(平衡因子)只能是-1、0或1。
- 旋转操作:在插入或删除节点后,如果某个节点的平衡因子超出范围,需要通过单旋转或双旋转来恢复平衡。
- 优点:查找操作效率高,适合频繁查找的场景。
- 缺点:插入和删除操作可能需要频繁的旋转,导致实现较为复杂。
2. 红黑树(Red-Black Tree)
红黑树是一种近似平衡的二叉搜索树,通过在每个节点上增加颜色属性(红色或黑色)来维持平衡。其特点包括:
- 性质:
- 每个节点要么是红色,要么是黑色。
- 根节点是黑色。
- 所有叶子节点(NIL节点)都是黑色。
- 如果一个节点是红色的,则其子节点必须是黑色的(即不能有两个连续的红色节点)。
- 从任一节点到其所有后代叶子节点的简单路径上,均包含相同数目的黑色节点。
- 旋转和颜色翻转:通过旋转和颜色翻转操作来维持红黑树的性质,从而保证树的平衡。
- 优点:插入和删除操作相对简单且高效,广泛应用于各种编程语言的标准库中,如C++的
std::map
和std::set
。 - 缺点:查找效率略低于AVL树,但在大多数应用中差异不大。
3. Splay树(伸展树)
Splay树是一种自调整的二叉搜索树,通过将最近访问的节点“伸展”到根部来保持树的局部性。其特点包括:
- 自调整:每次访问节点后,通过一系列旋转操作将该节点移至根部。
- 摊还复杂度:尽管单次操作的时间复杂度可能较高,但经过多次操作后的平均时间复杂度为对数级别。
- 优点:对于频繁访问某些节点的场景非常高效。
- 缺点:在最坏情况下,单次操作的时间复杂度为线性级别。
平衡二叉树的应用
- 数据库索引:平衡二叉树用于实现高效的索引结构,支持快速的数据检索、插入和删除。
- 内存管理:操作系统中的内存分配器使用平衡树来管理空闲内存块,确保快速分配和回收内存。
- 编程语言的标准库:许多编程语言的标准库使用平衡二叉树来实现关联容器,如C++的
std::map
、Java的TreeMap
等。 - 文件系统:一些文件系统使用平衡树来管理文件目录结构,确保快速的文件查找和操作。
总结
平衡二叉树通过保持树的高度在对数级别,从而确保各种操作的高效性。在实际应用中,根据具体需求和场景,可以选择不同类型的平衡二叉树,如AVL树适合查找频繁的场景,红黑树则在插入和删除操作频繁且对实现复杂度有要求的情况下表现更佳。理解平衡二叉树的原理和特点,有助于在设计高效数据结构和算法时做出更合适的选择。
我找的定义:左右子树高度相差不超过1,对于每一个节点。
代码
太蠢了,连抄代码都能抄错,问AI也打不出个所以然,下次在问AI前试着自己找出错误,最难就是逻辑上的。
class IsBalance
{
private:
bool balance;
public:
bool isBalanced(Node* root)
{
balance = true;
//balance是全局变量,所有调用过程共享
//每次开始设置为true
height(root);
return balance;
}
//对于判断平衡二叉树,一旦发现不平衡的节点,balance改为false并返回。
//height函数的作用:求当前以cur为根节点的子树的高度;判断是否平衡。
//一旦发现不平衡,返回什么高度就不重要了。
int height(Node* cur)
{
if (!balance || cur == NULL) return 0;
int left_height = height(cur->left);
int right_height = height(cur->right);
if (abs(left_height - right_height) > 1)
{
balance = false;
}
//发现不平衡后依旧正常返回
return left_height > right_height ? left_height + 1 : right_height + 1;
}
};
最sb的是我||写成&&。
五,验证搜索二叉树(树型dp沾边)
首先想到的就是中序遍历,看遍历结果是不是一直升序。
1,递归方式实现中序遍历,顺便验证是否是搜索二叉树
bool Is_BST(Node* root)
{
stack<int> stack;
midOrderTraverse(root, stack);
int pre = stack.top(), cur;
stack.pop();
while (!stack.empty())
{
cur = stack.top();
stack.pop();
if (cur > pre) return false;
pre = cur;
}
return true;
}
void midOrderTraverse(Node* root, stack<int>& stack)
{
if (!root) return;
midOrderTraverse(root->left, stack);
stack.push(root->value);
midOrderTraverse(root->right, stack);
}
2,搜索二叉树构建(递归方法)
// 插入节点的辅助函数
TreeNode* insert(TreeNode* node, int val) {
if (node == nullptr) {
return new TreeNode(val);
}
if (val < node->val) {
node->left = insert(node->left, val);
} else if (val > node->val) {
node->right = insert(node->right, val);
}
return node;
}
确实AI写得好一些,
#include<iostream>
#include<vector>
struct Node
{
int value;
Node* left;
Node* right;
Node(int x) : value(x), left(NULL), right(NULL) {};
};
class SearchBinaryTree
{
private:
Node* root;
// 插入节点的辅助函数,将值插入到以 node 为根的子树中
Node* insert(Node* node, int value)
{
if (node == NULL) return new Node(value);
if (value < node->value)
{
node->left = insert(node->left, value);
}
else if (value > node->value)
{
node->right = insert(node->right, value);
}
return node;
}
// 中序遍历辅助函数,用于输出树的节点值
void midOrderTraverse(Node* root)
{
if (root == NULL) return;
midOrderTraverse(root->left);
std::cout << root->value << " ";
midOrderTraverse(root->right);
}
public:
SearchBinaryTree() :root(NULL) {};
// 插入元素到二叉搜索树中
void insert(int value)
{
root = insert(root, value);
}
// 对二叉搜索树进行中序遍历
void midOrderTraverse()
{
midOrderTraverse(root);
std::cout << std::endl;
}
};
int main()
{
std::vector<int> nums = { 1,3,5,7,9,2,4,6,8 };
SearchBinaryTree bst;
for (int num : nums)
{
bst.insert(num);
}
bst.midOrderTraverse();
return 0;
}
3,递归版判断(使用全局变量)
static int max;
static int min;
//这两个全局变量是用来表示子树所有节点中的最大/小值
//是在递归函数结束时,给父节点返回。让父节点知道左/右子树的最大/小值
//在判断完该节点后,重置这两个全局变量。
bool Is_SBT1(Node* node)
{
if (node == NULL)
{
min = std::numeric_limits<int>::max();
max = std::numeric_limits<int>::min();
//确保比较有效
return true;
}
bool lok = Is_SBT1(node->left);//左边是不是满足条件
int lmin = min;
int lmax = max;
bool rok = Is_SBT1(node->right);//右边是不是。
int rmin = min;
int rmax = max;
//更新此时的max和min
min = std::min(std::min(lmin, rmin), node->value);
max = std::max(std::max(lmax, rmax), node->value);
return lok && rok && lmax < node->value && node->value < rmin;
}
我觉得最神奇的是全局变量的使用。它们时刻都在被改变,只记录当时的最值。
六,修剪二叉树
修剪二叉树(Trim Binary Tree)是一种对二叉树进行操作的算法,通常应用于二叉搜索树(Binary Search Tree,BST),其目的是移除树中不符合特定范围条件的节点,使得最终得到的二叉树仅包含节点值在指定范围 [low, high]
内的节点。
修剪二叉树的基本原理
- 对于二叉搜索树,其性质是左子节点的值小于父节点的值,右子节点的值大于父节点的值。利用这个性质可以有效地修剪二叉搜索树。
修剪步骤
- 递归遍历:
- 通常使用递归的方式遍历二叉树,因为二叉树的结构非常适合递归操作。
- 从根节点开始,对于每个节点进行以下操作:
- 若节点的值小于范围的下限
low
,那么该节点及其左子树都将被排除,所以修剪后的结果是对其右子树进行修剪的结果。 - 若节点的值大于范围的上限
high
,那么该节点及其右子树都将被排除,所以修剪后的结果是对其左子树进行修剪的结果。 - 若节点的值在范围
[low, high]
内,则该节点保留,并且对其左右子树分别进行修剪。
- 若节点的值小于范围的下限
1,看完视频第一次尝试
Node* trimBST(Node* root, int low, int high)
{
if (root == NULL) return NULL;
if (root->value >= low && root->value <= high)
{
root->left = trimBST(root->left, low, high);
root->right = trimBST(root->right, low, high);
return root;
}
return NULL;
}
没有正确处理当节点值不在范围之内时的情况,我是直接舍弃。这样会导致二叉树不完整。
2,第二次尝试
Node* trimBST(Node* root, int low, int high)
{
if (root == NULL) return NULL;
if (root->value < low)
{
return trimBST(root->right, low, high);
}
else if (root->value > high)
{
return trimBST(root->left, low, high);
}
else
{
root->left = trimBST(root->left, low, high);
root->right = trimBST(root->right, low, high);
return root;
}
}
七,二叉树打家劫舍问题(树型dp沾边)
题目要求
题目描述:
有一个二叉树,每个节点代表一个房屋,节点的值表示房屋中的财富。你可以选择偷窃某个房屋,但你不能同时偷窃相邻的两个房屋(即不能偷窃父子节点)。现在,你要计算能偷窃的最大财富。你需要设计一个算法来找出最大的偷盗金额。
具体规则:
- 你不能偷窃父节点和子节点。
- 每个节点的值表示该节点所代表的房屋的财富。
- 你可以选择偷窃或不偷窃每个房屋,但相邻的两个房屋不能同时被偷窃。
任务:
给定一个二叉树,计算在遵守上述规则的情况下,你能偷窃的最大财富。
示例 1:
输入:
3
/ \
2 3
\ \
3 1
输出:
7
解释:
- 选择节点 3(根节点),然后选择节点 3(右子节点)。你不能选择节点 2 和节点 1,因为它们是相邻的。
- 偷窃的财富为 3 + 3 + 1 = 7。
示例 2:
输入:
3
/ \
4 5
/ \ \
1 3 1
输出:
9
解释:
- 选择节点 4 和节点 5,但不选择其子节点(1 和 3)。最终偷窃的财富为 4 + 5 = 9。
思路:
此题类似于动态规划或树的递归问题,我们可以通过递归来计算每个节点的最大偷窃金额。递归的核心思想是:
- 对于每个节点,我们有两种选择:
- 偷窃当前节点:那么它的子节点就不能被偷窃了,偷窃当前节点的最大值为
当前节点的值 + 左子树和右子树不偷窃时的最大值
。 - 不偷窃当前节点:那么可以偷窃左右子树的最大值。
- 偷窃当前节点:那么它的子节点就不能被偷窃了,偷窃当前节点的最大值为
- 最终结果是,对于每个节点,返回偷窃和不偷窃两种选择的最大值。
递归实现:
在递归函数中,每个节点返回两种情况:
- 偷窃该节点的最大值。
- 不偷窃该节点的最大值。
通过递归得到每个节点的最大偷窃值,最终返回树的根节点的偷窃结果。
代码实现:
#include <iostream>
#include <algorithm>
using namespace std;
// 树节点结构
struct Node {
int value;
Node* left;
Node* right;
Node(int val) : value(val), left(nullptr), right(nullptr) {}
};
class BSTMaxTheft {
public:
int maxStealedValue(Node* root);
private:
pair<int, int> recursion(Node* root); // 返回值为 pair,第一个是偷当前节点的最大值,第二个是不偷当前节点的最大值
};
// 计算最大偷盗值
int BSTMaxTheft::maxStealedValue(Node* root)
{
pair<int, int> result = recursion(root);
return std::max(result.first, result.second);
}
// 递归计算每个节点的最大偷盗值
pair<int, int> BSTMaxTheft::recursion(Node* root)
{
if (root == nullptr) {
return {0, 0}; // 当前节点为空,返回偷和不偷的最大值都是 0
}
// 递归计算左子树和右子树的最大偷盗值
pair<int, int> left = recursion(root->left);
pair<int, int> right = recursion(root->right);
// 当前节点被偷时:当前节点的值 + 左子树和右子树不能偷的最大值
int steal = root->value + left.second + right.second;
// 当前节点不被偷时:选择左子树和右子树的最大值
int notSteal = max(left.first, left.second) + max(right.first, right.second);
return {steal, notSteal};
}
// 主函数
int main()
{
// 创建一个二叉树
Node* root = new Node(3);
root->left = new Node(2);
root->right = new Node(3);
root->left->right = new Node(3);
root->right->right = new Node(1);
BSTMaxTheft bst;
int maxValue = bst.maxStealedValue(root);
cout << "Max stolen value: " << maxValue << endl; // 输出 7
return 0;
}
展开
解释:
-
recursion
函数:- 该函数接收一个节点作为参数,返回一个
pair<int, int>
,其中:first
是偷窃当前节点的最大值。second
是不偷窃当前节点的最大值。
- 递归的终止条件是遇到空节点,返回
(0, 0)
。
- 该函数接收一个节点作为参数,返回一个
-
maxStealedValue
函数:- 该函数调用
recursion
获取根节点的最大偷盗值,并返回偷和不偷的最大值。
- 该函数调用
测试结果:
对于输入:
3
/ \
2 3
\ \
3 1
输出:
Max stolen value: 7
对于输入:
3
/ \
4 5
/ \ \
1 3 1
输出:
Max stolen value: 9
利用全局变量和递归,在每一个节点讨论偷该节点和不偷该节点的情况。
原理
两种方法:
利用类成员变量起着全局变量的作用,实现更新。将 yes
和 no
的计算保留为成员变量,但在每个节点递归时,将递归结果传递给相应的成员变量。在这种实现方式中,我们将使用成员变量来存储“是否偷盗”以及“不偷盗”的最大价值。
直接返回结果,将递归函数的返回类型更改为 pair<int, int>
,其中 first
表示当前节点被偷时的最大价值,second
表示当前节点不被偷时的最大价值。
法一(自己写的)
代码
int BSTMaxTheft::maxStealedValue(Node* root)
{
recursion1(root);
return std::max(yes, no);
}
void BSTMaxTheft::recursion1(Node* root)
{
int leftYes, rightYes;
int leftNo, rightNo;
if (root == NULL)
{
yes = no = 0;
return;
}
recursion1(root->left);
//此时更新了成员变量yes和no
leftYes = yes;
leftNo = no;
recursion1(root->right);
rightYes = yes;
rightNo = no;
yes = root->value + leftNo + rightNo;
no = std::max(leftYes, leftNo) + std::max(rightYes, rightNo);
}
法二(自己写的)
int BSTMaxTheft::maxStealedValue2(Node* root)
{
std::pair<int, int> ans = recursion2(root);
return std::max(ans.first, ans.second);
}
std::pair<int, int> BSTMaxTheft::recursion2(Node* root)
{
if (root == NULL)
{
return { 0,0 };
}
std::pair<int, int> left = recursion2(root->left);
std::pair<int, int> right = recursion2(root->right);
int yes = root->value + left.second + right.second;
int no = std::max(left.first, left.second) + std::max(right.first, right.second);
return { yes,no };
}