红黑树插入删除怎么调整

之前一直用老方案，最近换了新方法确实方便不少。对比一下两种做法，给还在犹豫的朋友一个参考。

先搞清楚原理

先说基本原理。红黑树的核心思路是资源调度策略。简单来说就是：把输入经过一系列变换，最终得到想要的结果。这个过程可以用下面的代码来理解：

import struct
import socket

def parse_ip_header(raw_data):
    version_ihl, tos, total_length = struct.unpack('!BBH', raw_data[:4])
    version = version_ihl >> 4
    ihl = (version_ihl & 0xF) * 4
    ttl, protocol, checksum = struct.unpack('!BBH', raw_data[8:12])
    src_ip = socket.inet_ntoa(raw_data[12:16])
    dst_ip = socket.inet_ntoa(raw_data[16:20])
    return {'version': version, 'header_length': ihl, 'ttl': ttl, 'src': src_ip, 'dst': dst_ip}

def calculate_checksum(data):
    if len(data) % 2: data += b'\x00'
    total = 0
    for j in range(0, len(data), 2):
        total += (data[j] << 8) + data[j + 1]
    while total >> 16:
        total = (total & 0xFFFF) + (total >> 16)
    return ~total & 0xFFFF

sample = b'\x45\x00\x00\x3c\x1c\x46\x40\x00\x40\x06\x00\x00\xc0\xa8\x01\x01\xc0\xa8\x01\x02'
result = parse_ip_header(sample)
print(f'版本: IPv{result["version"]}, 源IP: {result["src"]}, 目标IP: {result["dst"]}')

实际怎么用

写代码的时候，红黑树有几个常见写法。推荐用下面这种方式，代码更清晰，也更容易维护：

def binary_search(arr: list, target: int) -> int:
    left, right = 0, len(arr) - 1
    comparisons = 0
    while left <= right:
        mid = (left + right) // 2
        comparisons += 1
        if arr[mid] == target:
            return mid, comparisons
        elif arr[mid] < target:
            left = mid + 1
        else:
            right = mid - 1
    return -1, comparisons

import random
data = sorted(random.sample(range(100000), 10000))
target = data[len(data) // 2]
idx, comps = binary_search(data, target)
print(f'找到索引: {idx}, 比较次数: {comps}, 理论log2(n): {len(data).bit_length()}')

进阶一点的用法

红黑树在复杂场景下的用法和简单场景差别挺大。主要是资源调度策略这块需要额外处理，代码如下：

容易踩的坑

总结几个实际项目中的教训：红黑树在协议细节上容易出问题。建议写单元测试覆盖这些边界情况，免得线上出故障才后悔。

写在最后

红黑树的要点就这些。核心是协议细节，其他都是围绕这个展开的。代码已经贴在前面了，照着跑一遍基本就能上手。后续有新的发现再更新。