mirror of
https://github.com/youngyangyang04/leetcode-master.git
synced 2025-07-09 03:34:02 +08:00
添加 0226.翻转二叉树.md Scala版本
This commit is contained in:
@ -820,5 +820,53 @@ func invertTree(_ root: TreeNode?) -> TreeNode? {
|
|||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
|
### Scala
|
||||||
|
|
||||||
|
深度优先遍历(前序遍历):
|
||||||
|
```scala
|
||||||
|
object Solution {
|
||||||
|
def invertTree(root: TreeNode): TreeNode = {
|
||||||
|
if (root == null) return root
|
||||||
|
// 递归
|
||||||
|
def process(node: TreeNode): Unit = {
|
||||||
|
if (node == null) return
|
||||||
|
// 翻转节点
|
||||||
|
val curNode = node.left
|
||||||
|
node.left = node.right
|
||||||
|
node.right = curNode
|
||||||
|
process(node.left)
|
||||||
|
process(node.right)
|
||||||
|
}
|
||||||
|
process(root)
|
||||||
|
root
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
广度优先遍历(层序遍历):
|
||||||
|
```scala
|
||||||
|
object Solution {
|
||||||
|
import scala.collection.mutable
|
||||||
|
def invertTree(root: TreeNode): TreeNode = {
|
||||||
|
if (root == null) return root
|
||||||
|
val queue = mutable.Queue[TreeNode]()
|
||||||
|
queue.enqueue(root)
|
||||||
|
while (!queue.isEmpty) {
|
||||||
|
val len = queue.size
|
||||||
|
for (i <- 0 until len) {
|
||||||
|
var curNode = queue.dequeue()
|
||||||
|
if (curNode.left != null) queue.enqueue(curNode.left)
|
||||||
|
if (curNode.right != null) queue.enqueue(curNode.right)
|
||||||
|
// 翻转
|
||||||
|
var tmpNode = curNode.left
|
||||||
|
curNode.left = curNode.right
|
||||||
|
curNode.right = tmpNode
|
||||||
|
}
|
||||||
|
}
|
||||||
|
root
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
-----------------------
|
-----------------------
|
||||||
<div align="center"><img src=https://code-thinking.cdn.bcebos.com/pics/01二维码一.jpg width=500> </img></div>
|
<div align="center"><img src=https://code-thinking.cdn.bcebos.com/pics/01二维码一.jpg width=500> </img></div>
|
||||||
|
Reference in New Issue
Block a user