【30天玩转python】高级数据结构

高级数据结构

在 Python 中,除了基础的列表、元组、字典和集合等数据结构之外,还有一些更复杂和高级的数据结构。这些数据结构在解决特定问题时能够提供更好的性能和更强的功能。本节将介绍一些常用的高级数据结构,包括堆、队列、双端队列、链表、树、图等,了解它们的基本概念以及在 Python 中的实现方式。


1. 堆(Heap)

堆是一种特殊的树形数据结构,满足堆属性。堆可以分为最小堆最大堆,最小堆中父节点的值总是小于或等于子节点,而最大堆中父节点的值总是大于或等于子节点。

在 Python 中,heapq 模块提供了堆的实现,默认是最小堆。

1.1 使用 heapq 模块
python 复制代码
import heapq

# 创建一个最小堆
heap = []
heapq.heappush(heap, 10)
heapq.heappush(heap, 1)
heapq.heappush(heap, 5)

print(heap)  # 输出:[1, 10, 5]

# 弹出最小元素
min_element = heapq.heappop(heap)
print(min_element)  # 输出:1
1.2 堆排序

利用堆的性质可以实现高效的排序。

python 复制代码
def heap_sort(arr):
    heap = []
    for element in arr:
        heapq.heappush(heap, element)
    return [heapq.heappop(heap) for _ in range(len(heap))]

arr = [3, 1, 4, 1, 5, 9]
sorted_arr = heap_sort(arr)
print(sorted_arr)  # 输出:[1, 1, 3, 4, 5, 9]

2. 队列(Queue)

队列是一种先进先出(FIFO)的数据结构,最先加入的元素最先被移出。在 Python 中,可以使用 collections.deque 实现队列,也可以使用 queue.Queue 实现线程安全的队列。

2.1 使用 collections.deque
python 复制代码
from collections import deque

queue = deque([1, 2, 3])
queue.append(4)  # 入队
print(queue)  # 输出:deque([1, 2, 3, 4])

queue.popleft()  # 出队
print(queue)  # 输出:deque([2, 3, 4])
2.2 使用 queue.Queue
python 复制代码
from queue import Queue

q = Queue()
q.put(1)  # 入队
q.put(2)
q.put(3)

print(q.get())  # 出队,输出:1
print(q.get())  # 输出:2

3. 双端队列(Deque)

双端队列(Deque)是可以在两端进行插入和删除的队列。在 Python 中,可以使用 collections.deque 来实现双端队列。

python 复制代码
from collections import deque

dq = deque([1, 2, 3])
dq.append(4)       # 从右端插入
dq.appendleft(0)   # 从左端插入
print(dq)          # 输出:deque([0, 1, 2, 3, 4])

dq.pop()           # 从右端删除
dq.popleft()       # 从左端删除
print(dq)          # 输出:deque([1, 2, 3])

4. 链表(Linked List)

链表是一种线性数据结构,由多个节点组成。每个节点包含两部分:数据和指向下一个节点的引用。链表的优势是插入和删除元素非常高效,尤其是与数组相比。

4.1 单链表的实现
python 复制代码
class Node:
    def __init__(self, data):
        self.data = data
        self.next = None

class LinkedList:
    def __init__(self):
        self.head = None

    def append(self, data):
        new_node = Node(data)
        if not self.head:
            self.head = new_node
            return
        last_node = self.head
        while last_node.next:
            last_node = last_node.next
        last_node.next = new_node

    def print_list(self):
        current = self.head
        while current:
            print(current.data, end=" -> ")
            current = current.next
        print("None")

# 使用链表
ll = LinkedList()
ll.append(1)
ll.append(2)
ll.append(3)
ll.print_list()  # 输出:1 -> 2 -> 3 -> None

5. 树(Tree)

树是一种层次型的数据结构,由节点组成,通常包括一个根节点和若干子节点。树的常见类型有二叉树、二叉搜索树、平衡树等。

5.1 二叉树的实现
python 复制代码
class TreeNode:
    def __init__(self, data):
        self.data = data
        self.left = None
        self.right = None

# 创建二叉树
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)

# 前序遍历
def preorder_traversal(node):
    if node:
        print(node.data, end=" ")
        preorder_traversal(node.left)
        preorder_traversal(node.right)

preorder_traversal(root)  # 输出:1 2 4 5 3

6. 图(Graph)

图是由顶点和边组成的复杂数据结构,图可以是有向图或无向图,也可以是带权图。常用的表示方法包括邻接矩阵和邻接表。

6.1 使用邻接表表示图
python 复制代码
class Graph:
    def __init__(self):
        self.graph = {}

    def add_edge(self, u, v):
        if u not in self.graph:
            self.graph[u] = []
        self.graph[u].append(v)

    def print_graph(self):
        for node in self.graph:
            print(f"{node} -> {self.graph[node]}")

# 创建图
g = Graph()
g.add_edge(0, 1)
g.add_edge(0, 2)
g.add_edge(1, 2)
g.add_edge(2, 0)
g.add_edge(2, 3)

g.print_graph()
# 输出:
# 0 -> [1, 2]
# 1 -> [2]
# 2 -> [0, 3]

7. 哈希表(Hash Table)

哈希表是一种基于键值对的数据结构,使用哈希函数将键映射到对应的值。Python 中的字典(dict)实际上是哈希表的实现。

7.1 使用字典
python 复制代码
hash_table = {}
hash_table["name"] = "Alice"
hash_table["age"] = 25

print(hash_table)  # 输出:{'name': 'Alice', 'age': 25}
print(hash_table["name"])  # 输出:Alice

8. 小结

高级数据结构为复杂问题的解决提供了灵活高效的工具。通过理解堆、队列、双端队列、链表、树、图和哈希表的实现和使用,程序员可以选择适合的问题模型,并提高代码的性能和可扩展性。在不同的应用场景下,选择合适的数据结构至关重要。

相关推荐
Dream_Snowar几秒前
速通Python 第三节
开发语言·python
菜鸡中的奋斗鸡→挣扎鸡1 小时前
滑动窗口 + 算法复习
数据结构·算法
蓝天星空1 小时前
Python调用open ai接口
人工智能·python
jasmine s1 小时前
Pandas
开发语言·python
郭wes代码1 小时前
Cmd命令大全(万字详细版)
python·算法·小程序
leaf_leaves_leaf2 小时前
win11用一条命令给anaconda环境安装GPU版本pytorch,并检查是否为GPU版本
人工智能·pytorch·python
夜雨飘零12 小时前
基于Pytorch实现的说话人日志(说话人分离)
人工智能·pytorch·python·声纹识别·说话人分离·说话人日志
404NooFound2 小时前
Python轻量级NoSQL数据库TinyDB
开发语言·python·nosql
天天要nx2 小时前
D102【python 接口自动化学习】- pytest进阶之fixture用法
python·pytest
minstbe2 小时前
AI开发:使用支持向量机(SVM)进行文本情感分析训练 - Python
人工智能·python·支持向量机