DSAA b BST &(balance)BST code analysis II(南科大)

文章介绍了如何使用红黑树的特性来解决二叉搜索树可能存在的不平衡问题,通过旋转和颜色翻转操作确保树的深度均衡,从而保持搜索的时间复杂度在理想范围内。主要涉及get、put、rotateLeft、rotateRight和flipColors等关键函数的实现。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

注:文章中代码来自于何老师的代码,本文意在对于代码有更深的了解的解释文章

在二叉搜索树中,我们经常可能会遇到树的大小不均等的问题,可能会对时间复杂度有较大的影响,这个时候可以使用红黑树的连接法,来保证二叉搜索树的深度均等,来保证搜索时间复杂度维持在一个比较理想的区间。

  1. 在此函数中,get函数与二叉搜索树相同不变

  1. public Value get (Key key) {
  2. Node x = root;
  3. while (x != null) {
  4. int cmp = key.compareTo(x.key);
  5. if (cmp < 0) x = x.left;
  6. else if (cmp > 0) x = x.right;
  7. else if (cmp == 0) return x.val;
  8. }
  9. return null;
  10. }
  1. root我们也需要进行一些改进,添加了color作为判断节点与父节点之间的关系(是否属于相同登记)使用BLACK and RED来代表红黑树的黑色和红色(代表在2—3树中等效),用来判断右旋和左旋以及变向的条件

  1. private class Node{
  2. private Key key;
  3. private Value val;
  4. private int count;
  5. private Node left,right;
  6. boolean color;
  7. public Node(Key key,Value val,boolean color){
  8. this.key=key;
  9. this.val=val;
  10. this.color=color;
  11. }
  12. }
  13. private Node rotateLeft(Node h){
  14. assert isRed(h.right);
  15. Node x=h.right;
  16. h.right=x.left;
  17. x.left=h;
  18. x.color=h.color;
  19. h.color=RED;
  20. return x;
  21. }
  22. private Node rotateRight(Node h){
  23. assert isRed(h.left);
  24. Node x=h.left;
  25. h.left=x.right;
  26. x.right=h;
  27. x.color=h.color;
  28. h.color=RED;
  29. return x;
  30. }
  31. private void flipColors(Node h) {
  32. assert !isRed( h);
  33. assert isRed( h.left);
  34. assert isRed( h.right);
  35. h.color = RED;
  36. h.left.color = BLACK;
  37. h.right.color = BLACK;
  38. }
  1. put函数的改变比较大,因为为了维持二叉搜索树的深度适当,可能需要不停变换根节点以及各个父节点,因此在put进去一个node的时候,将node的颜色变成红色,并且向上逐一判断是否需要进行左旋右旋以及翻转

  1. private Node put (Node h, Key key, Value val) {
  2. if (h == null) return new Node( key, val, RED);
  3. int cmp = key.compareTo( h.key);
  4. if (cmp < 0) h.left = put( h.left, key, val);
  5. else if (cmp > 0) h.right = put( h.right, key, val);
  6. else if (cmp == 0) h.val = val;
  7. if (isRed(h.right) && !isRed(h.left)) h = rotateLeft( h);
  8. if (isRed(h.left) && isRed(h.left.left)) h = rotateRight( h);
  9. if (isRed(h.left) && isRed(h.right)) flipColors( h);
  10. return h;
  11. }

代码总体实例:

  1. public class BST2 <Key extends Comparable<Key>, Value>{
  2. private static final boolean RED=true;
  3. private static final boolean BLACK=false;
  4. private Node root;
  5. private class Node{
  6. private Key key;
  7. private Value val;
  8. private int count;
  9. private Node left,right;
  10. boolean color;
  11. public Node(Key key,Value val,boolean color){
  12. this.key=key;
  13. this.val=val;
  14. this.color=color;
  15. }
  16. }
  17. public int size()
  18. { return size( root); }
  19. private int size(BST2.Node x){
  20. if (x==null) return 0;
  21. return x.count;
  22. }
  23. public Value get (Key key) {
  24. Node x = root;
  25. while (x != null) {
  26. int cmp = key.compareTo(x.key);
  27. if (cmp < 0) x = x.left;
  28. else if (cmp > 0) x = x.right;
  29. else if (cmp == 0) return x.val;
  30. }
  31. return null;
  32. }
  33. private boolean isRed(Node x){
  34. if (x==null) return false;
  35. return x.color==RED;
  36. }
  37. private Node rotateLeft(Node h){
  38. assert isRed(h.right);
  39. Node x=h.right;
  40. h.right=x.left;
  41. x.left=h;
  42. x.color=h.color;
  43. h.color=RED;
  44. return x;
  45. }
  46. private Node rotateRight(Node h){
  47. assert isRed(h.left);
  48. Node x=h.left;
  49. h.left=x.right;
  50. x.right=h;
  51. x.color=h.color;
  52. h.color=RED;
  53. return x;
  54. }
  55. private void flipColors(Node h) {
  56. assert !isRed( h);
  57. assert isRed( h.left);
  58. assert isRed( h.right);
  59. h.color = RED;
  60. h.left.color = BLACK;
  61. h.right.color = BLACK;
  62. }
  63. private Node put (Node h, Key key, Value val) {
  64. if (h == null) return new Node( key, val, RED);
  65. int cmp = key.compareTo( h.key);
  66. if (cmp < 0) h.left = put( h.left, key, val);
  67. else if (cmp > 0) h.right = put( h.right, key, val);
  68. else if (cmp == 0) h.val = val;
  69. if (isRed(h.right) && !isRed(h.left)) h = rotateLeft( h);
  70. if (isRed(h.left) && isRed(h.left.left)) h = rotateRight( h);
  71. if (isRed(h.left) && isRed(h.right)) flipColors( h);
  72. return h;
  73. }
  74. }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值

举报

选择你想要举报的内容(必选)
  • 内容涉黄
  • 政治相关
  • 内容抄袭
  • 涉嫌广告
  • 内容侵权
  • 侮辱谩骂
  • 样式问题
  • 其他
程序员都在用的中文IT技术交流社区

程序员都在用的中文IT技术交流社区

专业的中文 IT 技术社区,与千万技术人共成长

专业的中文 IT 技术社区,与千万技术人共成长

关注【CSDN】视频号,行业资讯、技术分享精彩不断,直播好礼送不停!

关注【CSDN】视频号,行业资讯、技术分享精彩不断,直播好礼送不停!

客服 返回顶部