2021.04.30DFS求好叶子节点对数量

2021.04.30DFS求好叶子节点对数量

（题目来源：https://leetcode-cn.com/problems/number-of-good-leaf-nodes-pairs/）

题目描述

给你二叉树的根节点 root 和一个整数 distance 。

如果二叉树中两个 叶 节点之间的 最短路径长度 小于或者等于 distance ，那它们就可以构成一组 好叶子节点对 。

返回树中 好叶子节点对的数量 。

思路

1. 在二叉树的题目中，首先考虑dfs。
2. dfs应该返回的信息：该节点下面叶子节点到当前位置的距离 和 个数。
3. 采用数组int[] cnt =new int[n+1]，cnt[i]表示叶子节点到当前位置距离为i的叶节点个数。

代码

	//自底向上的递归，用count数组统计和当前节点的距离为 i 的叶子节点数目，
	//多于distance的不用考虑
	int[] dfs(TreeNode cur, int distance){
		int[] cnt = new int[distance+1];
		if(cur == null) return cnt;
		if(cur.left == null && cur.right == null) {//其本身就是叶子节点
			//所以其父亲节点下 距其距离为1的节点个数有1个。
			cnt[1] = 1;
			return cnt;
		}
		int[] left = dfs(cur.left,distance);
		int[] right = dfs(cur.right,distance);
		for(int i = 1; i < distance; i++) {
			for(int j = 1; j <= distance-i; j++) {
				//求对数
				ans+=left[i]*right[j];
			}
		}
		//合并数组，求得当前节点的返回结果
		for(int i = 2; i <= distance; i++) {
			cnt[i] = left[i-1]+right[i-1];
		}
		return cnt;
	}

2021.6.1 再看此题

本题要求找到一棵树的叶子节点之间的关系，所以应该从叶子节点向根节点遍历。每次遍历都返回之前遍历状态。

本题中，返回的状态用int数组记录，为当前节点下的叶子节点的好叶子节点个数。如下图：

遍历A、B子树后，返回A、B子树的遍历状态和叶子节点的信息，此时开始遍历2号节点。遍历2号节点时，由于得到两树的状态信息，选择不再比较A，B树内部细节，而是比较A，B之间的叶子节点，看是否满足好叶子节点的关系。

遍历完2，3号节点后，开始遍历1号节点，此时也是如此。不再考虑颜色相同子树内部的信息（如果题目要求不同，也是可以考虑的），而是考虑蓝绿子树之间的信息。