[PYTHON] Laissez Code Day12 partir de zéro "617. Fusionner deux arbres binaires"

Aperçu

Il semble que des tests de codage soient menés à l'étranger lors d'entretiens d'ingénieurs, et dans de nombreux cas, l'essentiel est d'implémenter des fonctions et des classes spécifiques en fonction du thème.

En guise de contre-mesure, il semble qu'un site appelé Let Code prendra des mesures.

Un site qui forme une puissance algorithmique capable de résister aux tests de codage dont on parle très tôt.

Je pense qu'il vaut mieux avoir la puissance de l'algorithme d'un être humain, donc je vais résoudre le problème de manière irrégulière et écrire la méthode que j'ai pensé à ce moment-là sous forme de mémo.

Leetcode

Table de codes Leet commençant à zéro

Dernière fois Leet Code Day 11 à partir de zéro "1315. Somme des nœuds avec un grand-parent pair"

En gros, je voudrais résoudre l'acceptation facile par ordre décroissant.

problème

617. Merge Two Binary Trees

Le niveau de difficulté est facile et il y en a beaucoup de bons.

Le problème est d'ajouter les nombres pour chacune des deux dichotomies données et de renvoyer un nouvel arbre.

Comme mentionné ci-dessus, l'arbre auquel les valeurs numériques de chaque partie correspondante sont ajoutées est renvoyé en sortie.

Solution

L'idée est de renvoyer l'autre nœud si l'un des nœuds n'existe pas, «Aucun» si les deux nœuds n'existent pas, et d'ajouter les nombres si les deux nœuds existent.

C'est celui que j'ai écrit en détail pour le moment.

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def mergeTrees(self, t1: TreeNode, t2: TreeNode) -> TreeNode:
        if not t1 and not t2:
            return None
        if not t1:
            return t2
        if not t2:
            return t1
        if t1 and t2:
            ans = TreeNode(t1.val + t2.val)
            ans.left = self.mergeTrees(t1.left,t2.left)
            ans.right = self.mergeTrees(t1.right,t2.right)
            return ans
# Runtime: 92 ms, faster than 63.90% of Python3 online submissions for Merge Two Binary Trees.
# Memory Usage: 15.1 MB, less than 5.72% of Python3 online submissions for Merge Two Binary Trees.

Voici une réécriture pour essayer de simplifier un peu plus

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def mergeTrees(self, t1: TreeNode, t2: TreeNode) -> TreeNode:
        if not t1:
            return t2
        if not t2:
            return t1
        ans = TreeNode(t1.val + t2.val)
        ans.left = self.mergeTrees(t1.left,t2.left)
        ans.right = self.mergeTrees(t1.right,t2.right)
        return ans
        
# Runtime: 84 ms, faster than 94.01% of Python3 online submissions for Merge Two Binary Trees.
# Memory Usage: 14.9 MB, less than 5.72% of Python3 online submissions for Merge Two Binary Trees.

C'est plus rapide, quoique irrégulier.