Условие

Даны два бинарных дерева root1 и root2. Вы должны объединить два дерева в одно. Если оба узла находятся на одной позиции, их значения нужно сложить. Если узел существует только в одном дереве, использовать его как есть.

Примеры

Input: root1 = [1, 3, 2, 5], root2 = [2, 1, 3, null, 4, null, 7] Output: [3, 4, 5, 5, 4, null, 7]

Input: root1 = [1], root2 = [1, 2] Output: [2, 2]

Решение

/**
 * Example:
 * var ti = TreeNode(5)
 * var v = ti.`val`
 * Definition for a binary tree node.
 * class TreeNode(var `val`: Int) {
 *     var left: TreeNode? = null
 *     var right: TreeNode? = null
 * }
 */
 fun mergeTrees(root1: TreeNode?, root2: TreeNode?): TreeNode? {
    if (root1 == null) return root2 // Если первое дерево пустое, возвращаем второе
    if (root2 == null) return root1 // Если второе дерево пустое, возвращаем первое

    // Объединяем значения корневых узлов
    root1.`val` += root2.`val`

    // Рекурсивно объединяем левое и правое поддеревья
    root1.left = mergeTrees(root1.left, root2.left)
    root1.right = mergeTrees(root1.right, root2.right)

    return root1 // Возвращаем объединенное дерево
}

Временная сложность

O(n), где n — общее количество узлов в двух деревьях.

Пространственная сложность

O(h), где h — максимальная высота дерева, для стека вызовов.