二叉树（5）——链式二叉树

续上篇，我们继续讲解链式二叉树第K层的节点个数和查找值为x的节点的代码实现。

1 二叉树第K层的节点个数

思路分析

• 若为空，返回0
• 不为空，且k==1，返回1
• 不为空，且k>1，返回左子树的k-1层+右子树的k-1层

代码实现

int TreeLevelk(TreeNode* root, int k)
{
	assert(k > 0);
	if (root == NULL)
		return 0;
	if (k == 1)
		return 1;

	return TreeLevelk(root->left, k - 1) + TreeLevelk(root->right, k - 1);
}

 递归展开图

 2 查找值为x的节点

思路分析

• 若为空树，返回NULL
• 若根节点就是x，直接返回根节点
• 否则就进行递归，先找根节点的左子树，层层递归，若找到，就将地址返回调用它的函数，层层返回，直到返回到第一个调用的函数结束。

代码实现

// 二叉树查找值为x的结点
TreeNode* TreeFind(TreeNode* root, BTDataType x)
{
	if (root == NULL)
		return NULL;

	if (root->data == x)
		return root;

	TreeNode* ret1 = TreeFind(root->left, x);
	if (ret1)
		return ret1;

	TreeNode* ret2 = TreeFind(root->right, x);
	if (ret2)
		return ret2;

	return NULL;
}

int main()
{
	TreeNode* root = CreateTree();
	// 二叉树前序遍历
	PrevOrder(root);
	printf("\n");

    TreeNode* ret = TreeFind(root, 5);
	printf("TreeFind:%p\n", ret);
	ret->data++; //将5那个节点的值+1

	PrevOrder(root);
	printf("\n");

	return 0;
}

 

以下两种写法也可以

TreeNode* TreeFind(TreeNode* root, BTDataType x)
{
	if (root == NULL)
		return NULL;

	if (root->data == x)
		return root;

	TreeNode* ret1 = TreeFind(root->left, x);
	if (ret1)
		return ret1;

	return TreeFind(root->right, x);
}

bool TreeFind(TreeNode* root, BTDataType x)
{
	if (root == NULL)
		return NULL;

	if (root->data == x)
		return root;

	return TreeFind(root->left, x) || TreeFind(root->right, x);
}

 递归展开图

• 递归里 return并不是返回到最外面，而是回到调用它的地方，层层往上走！！
• 返回值一定要有变量去接受！
• NULL并不一定代表这个二叉树里就找不到x。