从根到叶的二进制数之和(Java)

题目

给出一棵二叉树,其上每个结点的值都是 0 或 1 。每一条从根到叶的路径都代表一个从最高有效位开始的二进制数。例如,如果路径为 0 -> 1 -> 1 -> 0 -> 1,那么它表示二进制数 01101,也就是 13 。 对树上的每一片叶子,我们都要找出从根到该叶子的路径所表示的数字。 以 10^9 + 7 为模,返回这些数字之和。
在这里插入图片描述

思想

递归,从根节点开始,num表示根节点到当前节点路径所表示的数字大小,num每递归一次,便乘以2并且加上当前节点的值,当递归到叶子结点之后,加入总和sum中。

代码

/** * Definition for a binary tree node. 
 * public class TreeNode { 
 * *     int val; 
 * *     TreeNode left; 
 * *     TreeNode right; 
 * *     TreeNode(int x) { val = x; } 
 * * } 
 * */
 class Solution {    
     static int sum;    
     public Solution(){        
         this.sum = 0;    
     }    
     public int sumRootToLeaf(TreeNode root) {        
         pre(root,0);       
         return sum;    
     }    
     public void pre(TreeNode root, int num){        
         if(root == null){            
             return;        
         }        
         num = num*2 + root.val;        
         if(root.left == null && root.right == null){            
             sum = sum + num;        
         }
         else{            
             pre(root.left,num);            
             pre(root.right,num);        
         }    
     }
}
发布了35 篇原创文章 · 获赞 2 · 访问量 952

猜你喜欢

转载自blog.csdn.net/y18771025420/article/details/103466037