623. Add One Row to Tree.go

package leetcode

import (
"github.com/halfrost/LeetCode-Go/structures"
)

// TreeNode define
typeTreeNode= structures.TreeNode

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 * Val int
 * Left *TreeNode
 * Right *TreeNode
 * }
 */
funcaddOneRow(root*TreeNode, vint, dint) *TreeNode {
ifd==1 {
tmp:=&TreeNode{Val: v, Left: root, Right: nil}
returntmp
 }
level:=1
addTreeRow(root, v, d, &level)
returnroot
}

funcaddTreeRow(root*TreeNode, v, dint, currLevel*int) {
if*currLevel==d-1 {
root.Left=&TreeNode{Val: v, Left: root.Left, Right: nil}
root.Right=&TreeNode{Val: v, Left: nil, Right: root.Right}
return
 }
*currLevel++
ifroot.Left!=nil {
addTreeRow(root.Left, v, d, currLevel)
 }
ifroot.Right!=nil {
addTreeRow(root.Right, v, d, currLevel)
 }
*currLevel--
}