LeetCode(563 二叉树坡度)

如题在这里插入图片描述
很显然需要用到递归,但是不适合从上至下穷举,那样对元素和的运算会有浪费。那么转换流程为从底部开始运算,计算节点坡度的同时缓存元素和

    static int num=0; //坡度
	
	public static int findTilt(TreeNode root) {
		Sum(root);	//包括自己在内的子元素之和
		return num;
	}
	
	static int Sum(TreeNode root) {
		if(root == null) {
			return 0;
		}
		int lsum=Sum(root.left);//左子树元素和
		int rsum=Sum(root.right);//右子树元素和
		num+=Math.abs(lsum-rsum);//加上当前左右子树差
		return root.val+lsum+rsum;
	}

提交时去掉static,很直观的写法,直接递归求子树元素和。
在这里插入图片描述
换一种写法,直接通过return控制递归findTilt方法

public static int findTilt1(TreeNode root) {
		if(root !=null) {
			if(root.left!=null) {
				findTilt1(root.left); //递归更新val及num 注意不能return
			}
			if(root.right!=null) {
				findTilt1(root.right);
			}
			root.val=root.val+(root.left==null?0:root.left.val)+(root.right==null?0:root.right.val);//更新节点元素为子树所有元素之和
			num+=Math.abs((root.left==null?0:root.left.val)-(root.right==null?0:root.right.val)); //更新num
		}
		return num;
	}

仅仅是写法不一样,结果是完全相同的

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值