【数据结构】二叉树经典入门算法题集锦

x33g5p2x  于2021-10-01 转载在 其他  
字(3.9k)|赞(0)|评价(0)|浏览(407)

前言:本章将通过六道来自LeetCode/牛客网中的二叉树相关算法题来介绍数据结构中二叉树在算法题中的应用。

1.单值二叉树

OJ链接
如果二叉树每个节点都具有相同的值,那么该二叉树就是单值二叉树。

只有给定的树是单值二叉树时,才返回 true;否则返回 false

思路分析:

我们知道“”具有传递性,如果ab&&bc,可以推出ac,一个父亲跟左右孩子的值相等

题解:
  1. bool isUnivalTree(struct TreeNode* root){
  2. if(root==NULL)
  3. {
  4. return true;
  5. }
  6. if(root->left&&root->left->val!=root->val) //如果左不为空,且左值等于根节点的值
  7. {
  8. return false;
  9. }
  10. if(root->right&&root->right->val!=root->val)
  11. {
  12. return false;
  13. }
  14. return isUnivalTree(root->left)&&isUnivalTree(root->right); //“==”具有传递性,神奇的递归
  15. }

2.二叉树的前序遍历

OJ链接
给你二叉树的根节点 root ,返回它节点值的 前序 遍历。

思路分析:

首先需要malloc一块内存给数组arr,数组arr用于保存二叉树前序遍历结果,这时候写个TreeSize求下二叉树节点个数。

然后写一个子函数,递归调用进行遍历(记住i需要传指针)。

题解:
  1. /** * Note: The returned array must be malloced, assume caller calls free(). */
  2. //输出型参数,不需要再手动进行打印
  3. int TreeSize(struct TreeNode*root) //求出二叉树中节点个数
  4. {
  5. if(root==NULL)
  6. {
  7. return 0;
  8. }
  9. return TreeSize(root->left)+TreeSize(root->right)+1;
  10. }
  11. void _preorderTraversal(struct TreeNode* root, int *arr,int* i)
  12. {
  13. if(root==NULL)
  14. {
  15. return;
  16. }
  17. arr[(*i)++]=root->val;
  18. _preorderTraversal(root->left,arr,i);
  19. _preorderTraversal(root->right,arr,i);
  20. }
  21. int* preorderTraversal(struct TreeNode* root, int* returnSize){
  22. *returnSize=TreeSize(root);
  23. int* arr=malloc(sizeof(int)* *returnSize);
  24. int i=0;
  25. _preorderTraversal(root,arr,&i);//这边i务必传指针,每个递归调用栈帧中都有一个i,传值的话下一层++i不会对上一层造成影响
  26. return arr;
  27. }

3.相同的树

OJ链接
给你两棵二叉树的根节点 pq ,编写一个函数来检验这两棵树是否相同。

如果两个树在结构上相同,并且节点具有相同的值,则认为它们是相同的。

思路分析:

分治:大问题分解成小问题,小问题再继续分解,直到分割成不可再分割的子问题。

考虑:如果都为空,结果为真;若一个为空,结果为假;若均不为空,但是节点数值不一样,结果为假。

最后递归调用(分治思想)。

题解:
  1. bool isSameTree(struct TreeNode* p, struct TreeNode* q){
  2. //判断可以直接获得真假结果的情况
  3. if(p==NULL&&q==NULL)
  4. {
  5. return true;
  6. }
  7. if(p==NULL||q==NULL)
  8. {
  9. return false;
  10. }
  11. if(p->val!=q->val)
  12. {
  13. return false;
  14. }
  15. return isSameTree(p->left,q->left)&&isSameTree(p->right,q->right);
  16. }

判断下上述算法的时间/空间复杂度?

4.对称二叉树

OJ链接
给定一个二叉树,检查它是否是镜像对称的。

思路分析:

先判断根节点是否为空,是就返回true,不是就对左右子树进行递归判断,而对左右子树进行递归就类似上一题了。

题解:
  1. bool _isSymmetric(struct TreeNode* root1,struct TreeNode* root2)
  2. {
  3. if(root1==NULL&&root2==NULL)
  4. {
  5. return true;
  6. }
  7. if(root1==NULL||root2==NULL)
  8. {
  9. return false;
  10. }
  11. if(root1->val!=root2->val)
  12. {
  13. return false;
  14. }
  15. return _isSymmetric(root1->left,root2->right) && _isSymmetric(root1->right,root2->left);
  16. }
  17. bool isSymmetric(struct TreeNode* root){
  18. if(root==NULL)
  19. {
  20. return true;
  21. }
  22. return _isSymmetric(root->left,root->right);
  23. }

5.另一棵子树

OJ链接
给你两棵二叉树 root 和 subRoot 。检验 root 中是否包含和 subRoot 具有相同结构和节点值的子树。如果存在,返回 true ;否则,返回 false 。

二叉树 tree 的一棵子树包括 tree 的某个节点和这个节点的所有后代节点。tree 也可以看做它自身的一棵子树。

思路分析:

先判断根节点是否为空,是则返回false;遍历root,拿到每个子树的跟,跟subRoot比较

以如下两颗数为例:

递归过程如下:

题解:
  1. bool isSameTree(struct TreeNode* p, struct TreeNode* q){
  2. //判断可以直接获得真假结果的情况
  3. if(p==NULL&&q==NULL)
  4. {
  5. return true;
  6. }
  7. if(p==NULL||q==NULL)
  8. {
  9. return false;
  10. }
  11. if(p->val!=q->val)
  12. {
  13. return false;
  14. }
  15. return isSameTree(p->left,q->left)&&isSameTree(p->right,q->right);
  16. }
  17. bool isSubtree(struct TreeNode* root, struct TreeNode* subRoot){
  18. //遍历root,拿到每个子树的跟,跟subRoot比较
  19. if(root==NULL)
  20. {
  21. return false;
  22. }
  23. if(isSameTree(root,subRoot))
  24. {
  25. return true;
  26. }
  27. return isSubtree(root->left,subRoot)||isSubtree(root->right,subRoot);
  28. }

6.二叉树遍历

OJ链接
描述

编一个程序,读入用户输入的一串先序遍历字符串,根据此字符串建立一个二叉树(以指针方式存储)。 例如如下的先序遍历字符串: ABC/#/#DE/#G/#/#F/#/#/# 其中“/#”表示的是空格,空格字符代表空树。建立起此二叉树以后,再对二叉树进行中序遍历,输出遍历结果。

输入描述:

输入包括1行字符串,长度不超过100。

输出描述:

可能有多组测试数据,对于每组数据, 输出将输入字符串建立二叉树后中序遍历的序列,每个字符后面都有一个空格。 每个输出结果占一行。

思路分析:

题干啥也没给,所以首先根据输入的字符串str[],“逆前序遍历”复原出原本二叉树结构,然后中序遍历打印该二叉树。

题解:
  1. #include<stdio.h>
  2. struct TreeNode
  3. {
  4. char val;
  5. struct TreeNode* left;
  6. struct TreeNode* right;
  7. };
  8. struct TreeNode* CreatTree(char* str,int* pi)
  9. {
  10. if(str[*pi]=='#')
  11. {
  12. (*pi)++;
  13. return NULL;
  14. }
  15. struct TreeNode*root=(struct TreeNode*)
  16. malloc(sizeof(struct TreeNode));
  17. root->val=str[(*pi)++];
  18. root->left=CreatTree(str, pi);
  19. root->right=CreatTree(str, pi);
  20. return root;
  21. }
  22. void InOrder(struct TreeNode* root)
  23. {
  24. if(root==NULL)
  25. {
  26. return;
  27. }
  28. InOrder(root->left);
  29. printf("%c ",root->val);
  30. InOrder(root->right);
  31. }
  32. int main()
  33. {
  34. char str[100];
  35. scanf("%s",str);
  36. int i=0;
  37. struct TreeNode* root=CreatTree(str,&i);
  38. InOrder(root);
  39. }

相关文章