JAVA程序设计：二叉树最大宽度（LeetCode：662）

给定一个二叉树，编写一个函数来获取这个树的最大宽度。树的宽度是所有层中的最大宽度。这个二叉树与满二叉树（full binary tree）结构相同，但一些节点为空。

每一层的宽度被定义为两个端点（该层最左和最右的非空节点，两端点间的null节点也计入长度）之间的长度。

示例 1:

输入: 

           1
         /   \
        3     2
       / \     \  
      5   3     9 

输出: 4
解释: 最大值出现在树的第 3 层，宽度为 4 (5,3,null,9)。
示例 2:

输入: 

          1
         /  
        3    
       / \       
      5   3     

输出: 2
解释: 最大值出现在树的第 3 层，宽度为 2 (5,3)。
示例 3:

输入: 

          1
         / \
        3   2 
       /        
      5      

输出: 2
解释: 最大值出现在树的第 2 层，宽度为 2 (3,2)。
示例 4:

输入: 

          1
         / \
        3   2
       /     \  
      5       9 
     /         \
    6           7
输出: 8
解释: 最大值出现在树的第 4 层，宽度为 8 (6,null,null,null,null,null,null,7)。
注意: 答案在32位有符号整数的表示范围内。

思路：记一下每层开始和结尾就好啦

class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x) { val = x; }
}

class Solution {
	
	int[] a;
	int[] b;
	
   public int widthOfBinaryTree(TreeNode root) {
	   
	   if(root==null) return 0;
	   if(root.left==null && root.right==null) 
		   return 1;
	   
       a=new int[10000];
       b=new int[10000];
       for(int i=0;i<10000;i++) a[i]=-1;
       
       dfs(root,0,0);
       
       int max=0;
       for(int i=0;i<10000;i++) 
    	   max=Math.max(max, b[i]-a[i]);
       return max;
   }
   private void dfs(TreeNode root,int depth,int sum) {
	   if(root==null) return;
	   if(a[depth]==-1) a[depth]=sum;
	   b[depth]=sum+1;
	   dfs(root.left,depth+1,sum*2);
	   dfs(root.right,depth+1,sum*2+1);
   }
}