栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Java

【算法】1315. 祖父节点值为偶数的节点和(多语言实现)

Java 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

【算法】1315. 祖父节点值为偶数的节点和(多语言实现)


文章目录
  • 1315. 祖父节点值为偶数的节点和:
  • 样例 1:
  • 提示:
  • 分析
  • 题解
    • rust
    • go
    • typescript
    • c
    • c++
    • java
    • python
  • 原题传送门:https://leetcode.cn/problems/sum-of-nodes-with-even-valued-grandparent/


1315. 祖父节点值为偶数的节点和:

给你一棵二叉树,请你返回满足以下条件的所有节点的值之和:

  • 该节点的祖父节点的值为偶数。(一个节点的祖父节点是指该节点的父节点的父节点。)

如果不存在祖父节点值为偶数的节点,那么返回 0 。

样例 1:

输入:
	root = [6,7,8,2,7,1,3,9,null,1,4,null,null,null,5]
	
输出:
	18
	
解释:
	图中红色节点的祖父节点的值为偶数,蓝色节点为这些红色节点的祖父节点。
提示:
  • 树中节点的数目在 1 到 1 0 4 10^4 104 之间。
  • 每个节点的值在 1 到 100 之间。

分析
  • 面对这道算法题目,二当家的陷入了沉思。
  • 遍历结构,一般就是使用循环或是递归,这里显然递归更加直观易懂,所以选择使用套娃递归大法。
  • 二叉树的关联是单向的,无法从一个节点关联到它的祖父节点,所以我们需要把祖父节点或者说祖父节点的信息传递到递归中。
  • 父亲节点将会是当前节点子节点的祖父节点,而且也无法从一个节点关联到它的父亲节点,所以父亲节点的信息也要一并传递到递归中。

题解 rust
// Definition for a binary tree node.
// #[derive(Debug, PartialEq, Eq)]
// pub struct TreeNode {
//   pub val: i32,
//   pub left: Option>>,
//   pub right: Option>>,
// }
//
// impl TreeNode {
//   #[inline]
//   pub fn new(val: i32) -> Self {
//     TreeNode {
//       val,
//       left: None,
//       right: None
//     }
//   }
// }
use std::rc::Rc;
use std::cell::RefCell;
impl Solution {
    pub fn sum_even_grandparent(root: Option>>) -> i32 {
        fn dfs(mut node: Option>>, parentIsEven: bool, grandparentIsEven: bool) -> i32 {
            match node {
                Some(node) => {
                    let mut ans = 0;
                    
                    let mut node = node.borrow_mut();
                    if grandparentIsEven {
                        ans += node.val;
                    }

                    ans += dfs(node.left.take(), node.val & 1 == 0, parentIsEven);
                    ans += dfs(node.right.take(), node.val & 1 == 0, parentIsEven);
                    
                    ans
                }
                _ => { 0 }
            }
        }

        dfs(root, false, false)
    }
}

go
func sumEvenGrandparent(root *TreeNode) int {
    var dfs func(*TreeNode, bool, bool) int
	dfs = func(node *TreeNode, parentIsEven bool, grandparentIsEven bool) int {
		if node == nil {
			return 0
		}
		
		ans := 0

		if grandparentIsEven {
			ans += node.Val
		}

		ans += dfs(node.Left, (node.Val & 1) == 0, parentIsEven)
		ans += dfs(node.Right, (node.Val & 1) == 0, parentIsEven)

		return ans
	}

	return dfs(root, false, false)
}

typescript

function sumEvenGrandparent(root: TreeNode | null): number {
    const dfs = (node: TreeNode, parentIsEven: Boolean, grandparentIsEven: Boolean): number => {
        if (node == null) {
            return 0;
        }

        let ans = 0;

        if (grandparentIsEven) {
            ans += node.val;
        }

        ans += dfs(node.left, (node.val & 1) == 0, parentIsEven);
        ans += dfs(node.right, (node.val & 1) == 0, parentIsEven);

        return ans;
    };

    return dfs(root, false, false);
};

c
int dfs(struct TreeNode* node, bool parentIsEven, bool grandparentIsEven) {
    if (node == NULL) {
        return 0;
    }
    
    int ans = 0;

    if (grandparentIsEven) {
        ans += node->val;
    }

    ans += dfs(node->left, (node->val & 1) == 0, parentIsEven);
    ans += dfs(node->right, (node->val & 1) == 0, parentIsEven);

    return ans;
}

int sumEvenGrandparent(struct TreeNode* root){
    return dfs(root, false, false);
}

c++
class Solution {
private:
    int dfs(TreeNode* node, bool parentIsEven, bool grandparentIsEven) {
        if (node == nullptr) {
            return 0;
        }
        
        int ans = 0;

        if (grandparentIsEven) {
            ans += node->val;
        }

        ans += dfs(node->left, (node->val & 1) == 0, parentIsEven);
        ans += dfs(node->right, (node->val & 1) == 0, parentIsEven);

        return ans;
    }
public:
    int sumEvenGrandparent(TreeNode* root) {
        return dfs(root, false, false);
    }
};

java
class Solution {
    public int sumEvenGrandparent(TreeNode root) {
        return dfs(root, false, false);
	}

	private int dfs(TreeNode node, boolean parentIsEven, boolean grandparentIsEvent) {
		if (node == null) {
			return 0;
		}

		int ans = 0;
		
		if (grandparentIsEvent) {
			ans += node.val;
		}

		ans += dfs(node.left, (node.val & 1) == 0, parentIsEven);
		ans += dfs(node.right, (node.val & 1) == 0, parentIsEven);

		return ans;
	}
}

python
# 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(node: TreeNode, parent_is_even: bool, grandparent_is_even: bool) -> int:
            if node is None:
                return 0
            ans = 0
            if grandparent_is_even:
                ans += node.val
            ans += dfs(node.left, (node.val & 1) == 0, parent_is_even)
            ans += dfs(node.right, (node.val & 1) == 0, parent_is_even)
            return ans
        return dfs(root, False, False)
        


原题传送门:https://leetcode.cn/problems/sum-of-nodes-with-even-valued-grandparent/

非常感谢你阅读本文~
欢迎【点赞】【⭐收藏】【评论】~
放弃不难,但坚持一定很酷~
希望我们大家都能每天进步一点点~
本文由 二当家的白帽子:https://le-yi.blog.csdn.net/ 博客原创~


转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/949401.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号