当前位置:   article > 正文

剑指Offer C++ --- 数组篇_右偏树

右偏树

唉,真的是没有办法啊,只要是稍微大一点的公司都是先笔试,再面试,笔试过不了,八股背的再多,理解的再怎么透彻也没有发挥空间,只能硬着头皮刷题,我对算法真的很抵触,很苦恼,为了工作不得不做,废话不多说,直接开搞!今日目标20个题!

1. 数组重复的数字

  1. int findRepeatNumber(vector<int>& nums)
  2. {
  3. unordered_map<int,bool> map;
  4. for(auto x : nums)
  5. {
  6. if(map[x])
  7. {
  8. return x;
  9. }
  10. map[x] = true;
  11. }
  12. return -1;
  13. }

利用哈希表,很容易!

map[x] 的位置没有元素时,容器会默认添加一个key = x,val = 0的元素

2.二维数组的查找

  1. bool findNumberIn2DArray(vector<vector<int>>& matrix, int target)
  2. {
  3. int i = matrix.size();
  4. int j = 0;
  5. while(i>=0 && j<matrix[0].size)
  6. {
  7. //去掉一列
  8. if(matrix[i][j] < target)
  9. {
  10. j++;
  11. }
  12. //去掉一行
  13. else if(matrix[i][j] > target)
  14. {
  15. i--;
  16. }
  17. else
  18. {
  19. return false;
  20. }
  21. }
  22. return true;
  23. }

二分查找的变形,从左下角开始,此元素大于target,向上走一行,此元素小于target向右走一列

3.BST树:非递归删除

如果通过二分查找,找到待删除结点和其双亲节点

分为:三种情况

第一种:待删除结点没有孩子结点,直接将其双亲结点指向它的地址域置nullptr

第二种:待删除结点只有一个孩子结点,那就将其挂接在双亲结点指向待删除结点本身的地址域

第三种:待删除结点有两个孩子结点,找到待删除结点左子树中最大的结点或者右子树中最小的结点,将其data覆盖待删除结点的data,这样就转化为第一种或者第二种情况(即就是处理叶子结点)

特殊情况:对于左偏树,或者右偏树,删除根节点,需要重置根节点为其第一个非空孩子结点

  1. bool n_remove(Node& val)
  2. {
  3. Node* cur = root_;
  4. Node* parent = nullptr;
  5. //空树直接返回
  6. if(root_ == nullptr)
  7. {
  8. return false;
  9. }
  10. while(cur != nullptr)
  11. {
  12. if(cur—>data < val.data)
  13. {
  14. parent = cur;
  15. cur = cur->right;
  16. }
  17. else if(cur->data > val.data)
  18. {
  19. parent = cur;
  20. cur = cur->left;
  21. }
  22. else
  23. {
  24. break; //找到待删除结点
  25. }
  26. }
  27. if(cur == nullptr)
  28. {
  29. return false;
  30. }
  31. //先处理第三种情况
  32. if(cur->left != nullptr && cur->right != nullptr)
  33. {
  34. //找到待删除结点左子树中最大的结点
  35. parent = cur;
  36. Node* pre = cur->left
  37. while(pre->right!=nullptr)
  38. {
  39. parent = pre;
  40. pre = pre->right;
  41. }
  42. cur->data = pre->data;
  43. cur = pre;
  44. }
  45. //合并只有一个孩子是左孩子,还是右孩子,或者没有孩子的情况
  46. Node* child = cur->left;
  47. if(child == nullptr)
  48. {
  49. child = cur->right;
  50. }
  51. //处理左偏树或者右偏树
  52. if(parent == nullptr)
  53. {
  54. root_ = child;
  55. }
  56. else
  57. {
  58. //用待删除结点的孩子结点,覆盖其在双亲结点的地址域
  59. if(parent->left == cur)
  60. {
  61. parent->left = child;
  62. }
  63. else
  64. {
  65. parent->right = child;
  66. }
  67. }
  68. delete cur;
  69. return true;
  70. }

4.BST树递归层序遍历

  1. //计算树高
  2. int heigh(Node* node)
  3. {
  4. if(node == nullptr)
  5. {
  6. return 0;
  7. }
  8. int left = heigh(node->left);
  9. int right = heigh(node->right);
  10. return left>right ? left+1 : right+1;
  11. }
  12. //计算树的结点个数
  13. int nums(Node* node)
  14. {
  15. if(node == nullptr)
  16. {
  17. return 0;
  18. }
  19. int left = nums(node->left);
  20. int right = nums(node->right);
  21. return left + right + 1;
  22. }
  23. void levelOrder(Node* node,int i)
  24. {
  25. if(node == nullptr)
  26. {
  27. return;
  28. }
  29. if(i == 0)
  30. {
  31. cout << node->data << " ";
  32. return;
  33. }
  34. levelOrder(node->left,i-1);
  35. levelOrder(node->right,i-1);
  36. }
  37. void levelOrder()
  38. {
  39. int n = heigh(node);
  40. //用i来控制递归的层数
  41. for(int i = 0;i<n;++i)
  42. {
  43. levelOrder(node,i);
  44. }
  45. cout << endl;
  46. }

5.BST树递归插入

  1. Node* insert(Node* node,int val)
  2. {
  3. if(node == nullptr)
  4. {
  5. return new Node(val);
  6. }
  7. //不插入相同的元素
  8. if(node->data == val)
  9. {
  10. return node;
  11. }
  12. else if(node->data < val)
  13. {
  14. node->right = insert(node->right,val);
  15. }
  16. else
  17. {
  18. node->left = insert(node->left,val);
  19. }
  20. return node;
  21. }

6.BST树的递归删除

  1. Node* remove(Node* node, int val)
  2. {
  3. if(node == nullptr)
  4. {
  5. return nullptr;
  6. }
  7. //找到待删除结点
  8. if(node->data == val)
  9. {
  10. //三种情况处理
  11. if(node->left != nullptr && node->right != nullptr)
  12. {
  13. Node* pre = node->left;
  14. //待删除结点左子树中最大的结点
  15. while(pre->right != nullptr)
  16. {
  17. pre = pre->right;
  18. }
  19. //递归直接删除---待删除结点
  20. node->data = pre->data;
  21. node->left = remove(node->left,pre->data);
  22. }
  23. else
  24. {
  25. //只有一个孩子结点
  26. if(node->left != nullptr)
  27. {
  28. Node* left = node->left;
  29. delete node;
  30. return left;
  31. }
  32. else if(node->right != nullptr)
  33. {
  34. Node* right = node->right;
  35. delete node;
  36. return right;
  37. }
  38. //没有孩子结点
  39. else
  40. {
  41. delete node;
  42. return nulllptr;
  43. }
  44. }
  45. }
  46. else if(node->data < val)
  47. {
  48. node->right = remove(node->right,val);
  49. }
  50. else
  51. {
  52. node->left = remove(node->left,val);
  53. }
  54. return node;
  55. }

7.BST树非递归前序遍历(单栈)

  1. void preOrder()
  2. {
  3. if(root_ == nullptr)
  4. {
  5. return nullptr;
  6. }
  7. stack<Node*> st;
  8. st.push(root_);
  9. while(!st.empty())
  10. {
  11. Node* node = st.top(); //v
  12. st.pop();
  13. cout << node->data << " ";
  14. if(node->right != nullptr)
  15. {
  16. st.push(node->right); //L
  17. }
  18. if(node->left != nullptr) //R
  19. {
  20. st.push(node->left);
  21. }
  22. }
  23. cout << endl;
  24. }

思路:每循环一次先将栈顶元素出栈,输出栈顶元素,再将栈顶元素的右孩子先入栈,再将其左孩子入栈

8.BST树的中序遍历(单栈)

  1. void inOrder()
  2. {
  3. if(root_ == nullptr)
  4. {
  5. return;
  6. }
  7. stack<Node*> st;
  8. Node* cur = root_;
  9. while(!st.empty() || cur != nullptr)
  10. {
  11. if(cur!=nullptr)
  12. {
  13. st.push(cur);
  14. cur = cur->left;
  15. }
  16. else
  17. {
  18. cur = st.top();
  19. st.pop();
  20. cout << cur->data << " ";
  21. cur = cur->right;
  22. }
  23. }
  24. cout << endl;
  25. }

思路:先将左侧的结点入到栈里,如果左侧结点遇到nullptr,弹出栈顶元素并输出,转而入弹出元素的右侧结点

9.BST树的后序遍历 (双栈)

思路:将LRV 转化为 VRL 输出的时候又转化为 LRV

利用栈1存储 LRV, 按顺序转入栈2后,变为 VRL

  1. void postOrder()
  2. {
  3. if(root_ == nullptr)
  4. {
  5. return;
  6. }
  7. stack<Node*> st1;
  8. stack<Node*> st2;
  9. st1.push(root_);
  10. while(!st1.empty())
  11. {
  12. Node* node = st1.top();
  13. st1.pop();
  14. st2.push(node);
  15. if(node->left != nullptr)
  16. {
  17. st1.push(node->left);
  18. }
  19. if(node->right != nullptr)
  20. {
  21. st2.push(node->right);
  22. }
  23. }
  24. while(!st2.empty)
  25. {
  26. cout <<st2.top()->data << " ";
  27. st2.pop();
  28. }
  29. cout << endl;
  30. }

10.BST树的广度优先遍历

  1. void levelOrder()
  2. {
  3. if(root_ == nullptr)
  4. {
  5. return nullptr;
  6. }
  7. queue<Node*> q;
  8. q.push(root_);
  9. while(!q.empty())
  10. {
  11. Node* node = q.front();
  12. q.pop();
  13. cout << node->data << " ";
  14. if(node->left!=nullptr)
  15. {
  16. q.push(node->left);
  17. }
  18. if(node->right!=nullptr)
  19. {
  20. q.push(node->right);
  21. }
  22. }
  23. cout << endl;
  24. }

11.已知二叉树的前序中序遍历,重建二叉树

思路:采取分治的思想

 

i与j之间的距离要与m到n之间的距离一致

先序遍历第一个结点为根节点

在中序遍历中找到根节点,从第一个结点到根结点(不含根节点)全部都是左子树序列,从根节点(不含根节点)到末尾全部都是右子树序列

对应到先序遍历的序列中,按照相对位置计算,根节点到左子树序列的最后一个结点的距离等于中序遍历左子树序列的长度

首先可以先确定根结点的位置,那么下一个左子树的结点一定在先序序列根节点的下一个位置,对应到中序遍历应该是在中序遍历开始到根节点的位置-1之间,相应的就可以缩减先序序列的范围,起始位置+1到末尾距离要等于中序遍历开始到根节点的位置-1的距离

先序序列和中序序列,只要其中有一个错开,表示范围内没有合适的元素了

因为在中序遍历中下一个根节点一定在上一个根节点的左侧,而先序序列查找的范围距离要与中序序列查找范围的距离相等

  1. TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder)
  2. {
  3. return rebuild(preorder,0,preorder.size()-1,inorder,0,inorder.size()-1);
  4. }
  5. TreeNode* rebuild(vector<int>& preorder,int i,int j,vector<int>& inorder,int m,int n)
  6. {
  7. if(i>j || m>n)
  8. {
  9. return nullptr;
  10. }
  11. TreeNode *node = new TreeNode(preorder[i]);
  12. for(int k = m; k<=n; k++)
  13. {
  14. node->left = rebuild(preorder,i+1,i+k-m,inorder,m,k-1);
  15. node->right = rebuild(preorder,i+k-m+1,j,inorder,k+1,n);
  16. return node;
  17. }
  18. return node;
  19. }

12.旋转数组的最小数字

O(n)复杂度

思路:可看作两个分别递增的数组

 

  1. int minArray(vector<int>& numbers)
  2. {
  3. int i = numbers.size()-1;
  4. if(numbers[0] < numbers[i])
  5. {
  6. return numbers[0];
  7. }
  8. while(i>0)
  9. {
  10. if(numbers[i-1]<numbers[i])
  11. {
  12. --i;
  13. }
  14. else
  15. {
  16. return numbers[i];
  17. }
  18. }
  19. return numbers[i];
  20. }

O(logn)

m = (i + j) / 1;

nums[m] > nums[j]    旋转点在【m+1,j】  i = m+1

nums[m] < nums[j]   旋转点【i,m】   j = m

nums[m] == nums[j]  不确定     j =  j-1

  1. int minArray(vector<int>& numbers)
  2. {
  3. int i = 0;
  4. int j = numbers.size()-1;
  5. while(i < j)
  6. {
  7. int m = i+(j-i)/2;
  8. if(numbers[m] < numbers[j])
  9. {
  10. j = m;
  11. }
  12. else if(numbers[m] > numbers[j])
  13. {
  14. i = m+1;
  15. }
  16. else
  17. {
  18. j -= 1;
  19. }
  20. }
  21. return numbers[i];
  22. }

 当范围只剩下一个元素时,即是最小值

很遗憾今天只写了12道,明天继续加油吧!

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/article/detail/43705
推荐阅读
相关标签
  

闽ICP备14008679号