每日一题 2415. 反转二叉树的奇数层(中等,树)
2023-12-16 00:51:32
- 深度搜索整棵树,将奇数层的节点按照层数存在list中
- 将每一层 list 中的节点的值反转
- 这适用于所有二叉树,但题中的完美二叉树有更好的解法
- 我们可以把 root.left 和 root.right 看成一对节点,首先它们本身是需要反转的,然后下一层的 root.left.left 和 root.right.right,root.left.right 和 root.right.left 可以变成两对,且如果需要反转的话,只需要将每一对反转即可
- 可以看到,每一对节点传递到下一层可以变成两对节点,由于它是完美二叉树,这样的过程可以传递到最后一层,只需要将奇数层的节点对的值进行互换即可
class Solution:
def reverseOddLevels(self, root: Optional[TreeNode]) -> Optional[TreeNode]:
d = defaultdict(list)
def dfs(node, deep):
if node is None:
return
if deep % 2 == 1:
d[deep].append(node)
dfs(node.left, deep + 1)
dfs(node.right, deep + 1)
dfs(root, 0)
i = 1
while True:
if i not in d:
break
n = len(d[i])
for k in range(n // 2):
d[i][k].val, d[i][n - k - 1].val = d[i][n - k - 1].val, d[i][k].val
i += 2
return root
class Solution {
public:
TreeNode* reverseOddLevels(TreeNode* root) {
dfs(root->left, root->right, true);
return root;
}
void dfs(TreeNode *root1, TreeNode *root2, bool isOdd) {
if (root1 == nullptr) {
return;
}
if (isOdd) {
swap(root1->val, root2->val);
}
dfs(root1->left, root2->right, !isOdd);
dfs(root1->right, root2->left, !isOdd);
}
};
作者:力扣官方题解
链接:https://leetcode.cn/problems/reverse-odd-levels-of-binary-tree/solutions/2562073/fan-zhuan-er-cha-shu-de-qi-shu-ceng-by-l-n034/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
文章来源:https://blog.csdn.net/qq_46636391/article/details/135014745
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。 如若内容造成侵权/违法违规/事实不符,请联系我的编程经验分享网邮箱:veading@qq.com进行投诉反馈,一经查实,立即删除!
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。 如若内容造成侵权/违法违规/事实不符,请联系我的编程经验分享网邮箱:veading@qq.com进行投诉反馈,一经查实,立即删除!