[PYTHON] Let Code Day 11 À partir de zéro "1315. Somme des nœuds avec un grand-parent pair"

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

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

Dernière fois Leet Code Day10 à partir de zéro "1431. Enfants avec le plus grand nombre de bonbons"

problème

1315. Sum of Nodes with Even-Valued Grandparent

Le niveau de difficulté est moyen.

Étant donné une dichotomie, vous devez renvoyer le parent de ce parent, la valeur totale des nœuds dont les grands-parents ont une valeur paire. S'il n'existe pas, "0" est renvoyé.

Input: root = [6,7,8,2,7,1,3,9,null,1,4,null,null,null,5] Output: 18 Explanation: The red nodes are the nodes with even-value grandparent while the blue nodes are the even-value grandparents.

Constraints: The number of nodes in the tree is between 1 and 10^4. The value of nodes is between 1 and 100.

Un exemple est comme celui-ci, avec «6» et «8» étant la condition d'être grands-parents et même, et «[2,7,1,3]» étant le petit-enfant de 6. La valeur totale de «[5]», qui est le petit-fils de 8, «18» est renvoyée.

Solution

Quand j'ai vu le problème, j'ai eu l'idée que le nœud actuel, le nœud parent et le nœud grand-parent sont maintenus pendant le glissement, et si le nœud grand-parent existe et sa valeur est divisée par 2, le reste est 0. Par exemple, l'idée est d'ajouter le nœud actuel à la valeur totale. Ce qui suit est une mise en œuvre de cette idée.

# 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:
    ans = 0
    def sumEvenGrandparent(self, root: TreeNode) -> int:
        def dfs(cur:TreeNode,par:TreeNode,gra:TreeNode):
            if gra and gra.val % 2 == 0:
                self.ans += cur.val
            if cur.left:
                dfs(cur.left,cur,par)
            if cur.right:
                dfs(cur.right,cur,par)

        dfs(root,None,None)
        return self.ans
# Runtime: 92 ms, faster than 98.47% of Python3 online submissions for Sum of Nodes with Even-Valued Grandparent.
# Memory Usage: 17.3 MB, less than 100.00% of Python3 online submissions for Sum of Nodes with Even-Valued Grandparent.

Cette fois, la façon de penser semblait efficace, je l'ai donc rendue très simple. C'est assez rapide et je pense avoir une assez bonne réponse.

Certaines personnes peuvent être curieuses de connaître la position de ans, mais en Python, il n'est pas possible de faire référence à une variable globale à mi-chemin dans une fonction et de la redéfinir en une variable locale à mi-chemin, et elle est locale quelque part dans la fonction. Une variable définie comme une variable est traitée comme une variable locale dès le début lorsqu'elle entre dans la fonction. Donc ça ressemble à ça.

C'était un peu désagréable, alors je l'ai recherché et j'ai trouvé que Python avait une variable «non locale». Lors de la définition d'une fonction à l'intérieur d'une fonction, elle semble être utilisée pour changer les variables de la fonction externe à partir de la fonction interne, et elle peut être écrite en utilisant ceci.

# 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 sumEvenGrandparent(self, root: TreeNode) -> int:
        def dfs(cur:TreeNode,par:TreeNode,gra:TreeNode):
            if gra and gra.val % 2 == 0:
                nonlocal ans
                ans += cur.val
            if cur.left:
                dfs(cur.left,cur,par)
            if cur.right:
                dfs(cur.right,cur,par)
        ans = 0
        dfs(root,None,None)
        return ans
# Runtime: 92 ms, faster than 98.47% of Python3 online submissions for Sum of Nodes with Even-Valued Grandparent.
# Memory Usage: 17.3 MB, less than 100.00% of Python3 online submissions for Sum of Nodes with Even-Valued Grandparent.

La vitesse ne change pas, et celle-ci est plus propre et je pourrais personnellement l'aimer.