leetcode-437-路径总和 III-java

题目及测试

package pid437;


/* 437. 路径总和 III

给定一个二叉树，它的每个结点都存放着一个整数值。

找出路径和等于给定数值的路径总数。

路径不需要从根节点开始，也不需要在叶子节点结束，但是路径方向必须是向下的（只能从父节点到子节点）。

二叉树不超过1000个节点，且节点数值范围是 [-1000000,1000000] 的整数。

示例：

root = [10,5,-3,3,2,null,11,3,-2,null,1], sum = 8

      10
     /  \
    5   -3
   / \    \
  3   2   11
 / \   \
3  -2   1

返回 3。和等于 8 的路径有:

1.  5 -> 3
2.  5 -> 2 -> 1
3.  -3 -> 11



*/
public class main {
	
	public static void main(String[] args) {
		Object[] x=new Object[]{10,5,-3,3,2,null,11,3,-2,null,1,null,null,null,null};	
		BinaryTree tree=new BinaryTree(x);
		tree.printTree(tree.root);
		System.out.println("ito2="+8);
		test(tree.root,8);
		
		
	}
		 
	private static void test(TreeNode ito,int ito2) {
		Solution solution = new Solution();
		int rtn;		
		long begin = System.currentTimeMillis();
		rtn = solution.pathSum(ito,ito2);//执行程序
		long end = System.currentTimeMillis();		
		System.out.println("rtn=" );
		System.out.println(rtn);
		System.out.println();
		System.out.println("耗时：" + (end - begin) + "ms");
		System.out.println("-------------------");
	}

}

没想出来

解法1（别人的）

遍历每个节点。 关键点：递归

计算以当前节点为路径终点的所有路径和。 关键点：用一个数组保存从根节点到当前节点路径

以当前结点为最终叶子结点向上追溯，路径上的任一结点为根节点到当前结点的路径和为sum的路径个数。 然后以同样的方法以自己的孩子结点为最终叶子结点计算。

就是sum代表要得到的和，array表示到从真root到该节点的所有节点。计算该节点，该节点+上级节点，该节点+上级节点+...+root，即不断向上sum，计算其中是sum的个数，然后递归子节点。

public int pathSum(TreeNode root, int sum) {
        return pathSum(root, sum, new int[1000], 0);
    }

    public int pathSum(TreeNode root, int sum, int[] array/*保存路径*/, int p/*指向路径终点*/) {
        if (root == null) {
            return 0;
        }
        int tmp = root.val;
        int n = root.val == sum ? 1 : 0;
        for (int i = p - 1; i >= 0; i--) {
            tmp += array[i];
            if (tmp == sum) {
                n++;
            }
        }
        array[p] = root.val;
        int n1 = pathSum(root.left, sum, array, p + 1);
        int n2 = pathSum(root.right, sum, array, p + 1);
        return n + n1 + n2;
    }