Java实现求二叉树路径总和

标题：Java实现求二叉树路径总和

public class TestPathSum02 {
	/**
	 * 使用广度遍历，
	 * 
	 */
	public boolean pathSum(TreeNode p, int target) {
		if(p == null) {
			return false;
		}
		Queue<TreeNode> q = new LinkedList<>();
		q.offer(p);
		
		while(!q.isEmpty()) {
			int size = q.size();
			for(int i = 0; i < size; i++) {
				TreeNode node = q.poll();
				if(node.left != null) {
					node.left.val += node.val;
					q.offer(node.left);
				}
				if(node.right != null) {
					node.right.val += node.val;
					q.offer(node.right);
				}
				if(node.left == null && node.right == null) {
					if(node.val == target) {
						return true;
					}
				}
			}
		}
		
		return false;
	}
	
	/**
	 * 使用深度遍历，递归
	 */
	public boolean pathSum02(TreeNode p, int target, int sum) {
		if(p == null) {
			return false;
		}else {
			boolean res1 = this.pathSum02(p.left, target, sum + p.val);
			boolean res2 = this.pathSum02(p.right, target, sum + p.val);
			if(p.left == null && p.right == null) {
				if(sum + p.val == target) {
					return true;
				}
			}
			
			if(res1 || res2) {
				return true;
			}
			
			return false;
		}
		
		
		
	}
	
	/**
	 * 初始化一个tree
	 * 类广度遍历
	 * @param a
	 * @return
	 */
	public TreeNode initTree(Integer[] a) {
		if(a == null || a.length == 0) {
			return null;
		}
		
		int t = 0;
		TreeNode p = new TreeNode(a[t]);  //至少有一个元素
		Queue<TreeNode> q = new LinkedList<>();
		q.offer(p);
		
		while(!q.isEmpty()) {
			TreeNode node = q.poll();
			if(t + 1 == a.length) {  //先判断数组中是否还有下一个元素
				return p;
			}else {
				t++;
				if(a[t] == null) {  //若下一个元素为null，则不需要创建新的节点
					node.left = null;
				}else {
					node.left = new TreeNode(a[t]);
					q.offer(node.left);
				}
			}
			if(t + 1 == a.length) {
				return p;
			}else {
				t++;
				if(a[t] != null){  //上面的简写，a[t] == null，不需要再赋值
					node.right = new TreeNode(a[t]);
					q.offer(node.right);
				}
			}
		}
		
		return p;
		
	}
	
	
	
	@Test
	public void test() {
		System.out.println("使用init的");
		//中的方法 initTree（Character[]）对于参数（char[]）不适用
//		Integer[] a = new Integer[] {1, 2, 3, 4, 5, 9, 10, null, 6, 7, 8, null, null, null, 11};
		Integer[] a = new Integer[] {1, 2};
		
		//初始化TreeNode
		TreeNode p = this.initTree(a);
		
//		System.out.println("使用广度遍历");
//		System.out.println("isTrue:" + this.pathSum(p, 3));//更改了TreeNode
		
		System.out.println("使用递归");
		System.out.println("isTrue:" + this.pathSum02(p, 2, 0));  //测试的时候要将上面的广度遍历注释掉，否则会有问题
	}
}