2022/04/18

Leetcode Note: Go - Same Tree

Same Tree - LeetCode
https://leetcode.com/problems/same-tree/

  • Go 言語で取り組んだメモ

所感

  • p, q の全要素を確認して一致するか比較すれば良さそう

実装準備

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func isSameTree(p *TreeNode, q *TreeNode) bool {
    
}

Solution

Same Tree - LeetCode
https://leetcode.com/problems/same-tree/solution/

  • Approach 1: Recursion
    • 再帰的に全要素をチェックする
  • Approach 2: Iteration
    • 反復して全要素をチェックする

回答

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func isSameTree(p *TreeNode, q *TreeNode) bool {

    if p == nil && q == nil {
        return true
    }
    
    if p == nil || q == nil {
        return false
    }
    
    if p.Val != q.Val {
        return false
    }

    return isSameTree(p.Right, q.Right) && isSameTree(p.Left, q.Left)
}
  • 3つの if でエラーハンドリングというか、シンプルな組み合わせをチェック
  • あとは再帰的に isSameTree を呼び出す
    • 右は右と比較、左は左と比較
  • Solution に Python でのサンプルがあったので簡単に実装できた

See Also