文章目录
6、树(中等篇2)
6.1、94. 二叉树的中序遍历
给定一个二叉树,返回它的中序 遍历。
示例:
输入: [1,null,2,3]
输出: [1,3,2]
进阶: 递归算法很简单,你可以通过迭代算法完成吗?来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-inorder-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路解析:中序遍历递归实现很简单,下面题解代码主要是非递归的方式。
题解代码(C++):
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
vector<int> res;
if(root==NULL) return res;
stack<TreeNode*> s;
TreeNode* cur = root;
while(cur!=NULL||!s.empty()){
while(cur!=NULL){
s.push(cur);
cur = cur->left;
}
if(!s.empty()){
cur = s.top();s.pop();
res.push_back(cur->val);
cur = cur->right;
}
}
return res;
}
};
题解代码(Java):
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<Integer> inorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<Integer>();
if(root==null) return res;
Stack<TreeNode> s = new Stack<TreeNode>();
TreeNode cur = root;
while(cur!=null || !s.isEmpty()){
while(cur!=null){
s.push(cur);
cur =cur.left;
}
if(!s.isEmpty()){
cur = s.pop();
res.add(cur.val);
cur = cur.right;
}
}
return res;
}
}
题解代码(Python):
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def inorderTraversal(self, root: TreeNode) -> List[int]:
res = []
if root is None:
return res
s = []
cur = root
while cur is not None or len(s)>0:
while cur is not None:
s.append(cur)
cur = cur.left
if len(s)>0:
cur = s.pop()
res.append(cur.val)
cur = cur.right
return res
6.2、1261. 在受污染的二叉树中查找元素
给出一个满足下述规则的二叉树:
root.val == 0
如果 treeNode.val == x 且 treeNode.left != null,那么treeNode.left.val == 2 * x + 1
如果 treeNode.val == x 且 treeNode.right != null,那么 treeNode.right.val == 2 * x + 2 现在这个二叉树受到「污染」,所有的treeNode.val 都变成了 -1。请你先还原二叉树,然后实现 FindElements 类:
FindElements(TreeNode* root) 用受污染的二叉树初始化对象,你需要先把它还原。 bool find(int
target) 判断目标值 target 是否存在于还原后的二叉树中并返回结果。示例 1:
输入: [“FindElements”,“find”,“find”]
[[[-1,null,-1]],[1],[2]]
输出:[null,false,true]
解释: FindElements findElements = newFindElements([-1,null,-1]);
findElements.find(1); // return False
findElements.find(2); // return True
示例 2:
输入: [“FindElements”,“find”,“find”,“find”]
[[[-1,-1,-1,-1,-1]],[1],[3],[5]]
输出: [null,true,true,false]
解释:
FindElements findElements = new FindElements([-1,-1,-1,-1,-1]);
findElements.find(1); // return True
findElements.find(3); // return
True findElements.find(5); // return False
示例 3:
输入: [“FindElements”,“find”,“find”,“find”,“find”]
[[[-1,null,-1,-1,null,-1]],[2],[3],[4],[5]]
输出:
[null,true,false,false,true]
解释: FindElements findElements = new FindElements([-1,null,-1,-1,null,-1]); findElements.find(2); // returnTrue
findElements.find(3); // return False
findElements.find(4); //return False
findElements.find(5); // return True提示:
TreeNode.val == -1 二叉树的高度不超过 20 节点的总数在 [1, 104] 之间 调用 find() 的总次数在 [1, 104] 之间 0 <= target <= 106
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-elements-in-a-contaminated-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路解析:首先是要修复受污染的树,本质上修改结点的值,首先是把根节点值置为0,然后递归修复左右子树,如果左右孩子存在,则根据传入的参数修改值,如果左右孩子不存在,则返回。
修复完树之后是查找,在修复的时候就把新值加入到set集合中,调用find()函数时可以使用set集合查找的方法即可。
题解代码(C++):
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class FindElements {
public:
set<int> myset;
void dfs(TreeNode* root,int val){
if(root==NULL) return;
root->val=val;
myset.insert(root->val);
dfs(root->left,2*root->val+1);
dfs(root->right,2*root->val+2);
}
FindElements(TreeNode* root) {
root->val = 0;
dfs(root->left,2*root->val+1);
dfs(root->right,2*root->val+2);
}
bool find(int target) {
return myset.find(target)!=myset.end();
}
};
/**
* Your FindElements object will be instantiated and called as such:
* FindElements* obj = new FindElements(root);
* bool param_1 = obj->find(target);
*/
题解代码(Java):
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class FindElements {
Set<Integer> myset;
public void dfs(TreeNode root,int val){
if(root==null) return;
root.val = val;
myset.add(val);
dfs(root.left,2*val+1);
dfs(root.right,2*val+2);
}
public FindElements(TreeNode root) {
this.myset = new HashSet<>();
root.val = 0;
dfs(root.left,1);
dfs(root.right,2);
}
public boolean find(int target) {
return this.myset.contains(target);
}
}
/**
* Your FindElements object will be instantiated and called as such:
* FindElements obj = new FindElements(root);
* boolean param_1 = obj.find(target);
*/
题解代码(Python):
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class FindElements:
def __init__(self, root: TreeNode):
self.myset = set()
root.val=0
self.dfs(root.left,1)
self.dfs(root.right,2)
def dfs(self,root:TreeNode,val:int):
if root is None:
return None
root.val=val
self.myset.add(val)
self.dfs(root.left,2*val+1)
self.dfs(root.right,2*val+2)
def find(self, target: int) -> bool:
return target in self.myset
# Your FindElements object will be instantiated and called as such:
# obj = FindElements(root)
# param_1 = obj.find(target)
6.3、230. 二叉搜索树中第K小的元素
给定一个二叉搜索树,编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。
说明: 你可以假设 k 总是有效的,1 ≤ k ≤ 二叉搜索树元素个数。
示例 1:
输入: root = [3,1,4,null,2], k = 13 / \ 1 4 \ 2
输出: 1
示例2:
输入: root = [5,3,6,2,4,null,null,1], k = 35 / \ 3 6 / \ 2 4 / 1
输出: 3
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/kth-smallest-element-in-a-bst
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路解析:二叉搜索树的中序遍历是升序的结果,找到第k个访问的结点值即可,可以递归遍历,也可以非递归遍历。
题解代码(C++):
/**
* Definition for a binary tree node.
* struct