Python实现简单的数据结构

单链表

class SingleNode(object):
    """单个的节点"""
    def __init__(self, item):
        # item存放数据元素
        self.item = item
        # next是下一个节点的标识(指针)
        self.next = None


class SingleLinkList(object):
    """单链表"""
    # 空的单链表   self.__head = None
    # 非空的单链表 self.__head = node
    def __init__(self, node=None):
        self.__head = node

    def is_empty(self):
        """判断链表是否为空"""
        if self.__head == None:
            return True
        else:
            return False
        # return self.__head == None

    def length(self):
        """链表长度"""
        # 空链表的情况下
        # cur游标 指向首节点 用来遍历
        cur = self.__head   # None

        count = 0
        while cur != None:
            count += 1
            # 将游标后移动一位
            cur = cur.next
        return count

    def travel(self):
        """遍历整个链表"""
        cur = self.__head
        while cur != None:
            # 节点中的数据
            print(cur.item, end=' ')
            cur = cur.next
        print("")

    def append(self, item):
        """链表尾部添加元素"""
        # item 你要具体插入的数据
        node = SingleNode(item)
        # 考虑空链表
        if self.is_empty():
            self.__head = node
        else:
            cur = self.__head  # None  node
            # AttributeError: 'NoneType' object has no attribute 'next'
            while cur.next != None:
                # 找到最后一个节点
                cur = cur.next
            cur.next = node

    def add(self, item):
        """链表头部添加元素"""
        # item 你要具体插入的数据
        node = SingleNode(item)
        # 将新节点的链接区next指向头结点
        node.next = self.__head
        # 将链表的头指向新节点
        self.__head = node

    def insert(self, pos, item):        # insert(2,100)  insert(-1,100)  insert(10,100)
        """指定位置添加元素"""
        # item 你要具体插入的数据
        # 头部插入
        if pos <= 0:
            self.add(item)
        # 尾部插入
        elif pos > self.length()-1:
            self.append(item)
        else:
            # 指定位置插入
            node = SingleNode(item)
            pre = self.__head
            count = 0
            while count < (pos - 1):
                count += 1
                pre = pre.next

            node.next = pre.next
            pre.next = node

    def remove(self, item):
        """删除节点"""
        # remove(100)   [100,200,100] 删除一个100

        cur = self.__head       # 空链表 None
        pre = None

        while cur != None:
            # 找到了指定的元素
            if cur.item == item:
                # 删除第一个元素
                if cur == self.__head:
                    self.__head = cur.next
                else:
                    # 删除
                    pre.next = cur.next
                break
            else:
                # 继续移动
                pre = cur
                cur = cur.next


    def search(self, item):
        """查找节点是否存在"""
        cur = self.__head
        while cur != None:
            if cur.item == item:
                return True
            else:
                # 让游标继续执行
                cur = cur.next
        return False


s = SingleLinkList()

"""测试结果"""
print(s.is_empty())     # 单链表是否为空
print(s.length())       # 单链表的长度

# 向单链表尾部加五个数据
s.append(10)
s.append(23)
s.append(13)
s.append(5)
s.append(34)

print(s.is_empty())
print(s.length())

s.travel()      # 遍历单链表

# 向指定位置插入数据
s.insert(-1, 6)
s.insert(10, 77)
s.insert(2, 11)
s.insert(1, 33)
s.travel()

print(s.search(77))     # 查找数据77是否在单链表中

s.remove(33)    # 删除数据33
s.travel()      # 再次遍历单链表

单向循环链表

class Node(object):
    """节点"""
    def __init__(self, item):
        # item存放数据元素
        self.item = item
        # next是下一个节点的标识(指针)
        self.next = None


class SinCycLinkList(object):
    """单项循环链表"""
    # 空的单链表   self.__head = None
    # 非空的单链表 node.next = node
    def __init__(self, node=None):
        self.__head = None
        if node:
            node.next = node

    def is_empty(self):
        """判断链表是否为空"""
        if self.__head == None:
            return True
        else:
            return False

    def length(self):
        """返回链表的长度"""
        # 空链表的情况下
        if self.is_empty():
            return 0

        # cur游标 指向首节点 用来遍历
        cur = self.__head  # None
        # is 对象   == 数值是否相等
        count = 1
        while cur.next != self.__head:
            count += 1
            # 将游标后移动一位
            cur = cur.next
        return count

    def travel(self):
        """遍历"""
        # 空链表
        if self.is_empty():
            return None

        cur = self.__head
        while cur.next != self.__head:
            # 节点中的数据
            print(cur.item, end=' ')
            cur = cur.next
        # 退出循环的时候,cur指向尾结点,但是尾结点并没有打印
        print(cur.item)

    def add(self, item):
        """链表头部添加元素"""
        # item 你要具体插入的数据
        node = Node(item)

        # 空链表
        if self.is_empty():
            self.__head = node
            node.next = self.__head
        else:
            cur = self.__head   # None
            while cur.next != self.__head:
                cur = cur.next
            # 找到尾节点

            # 添加的节点指向原先的头节点
            node.next = self.__head
            # __head指向node
            self.__head = node
            # 尾结点指向新的节点
            cur.next = node

    def append(self, item):
        """链表尾部添加元素"""
        node = Node(item)
        # 空链表
        if self.is_empty():
            self.__head = node
            node.next = self.__head
        else:
            cur = self.__head
            while cur.next != self.__head:
                cur = cur.next
            # 找到尾节点

            # 将尾结点指向node
            cur.next = node
            # 将node指向__head
            node.next = self.__head

    def insert(self, pos, item):
        """指定位置添加元素"""
        # 头部插入
        if pos <= 0:
            self.add(item)
        # 尾部插入
        elif pos > self.length() - 1:
            self.append(item)
        else:
            # 指定位置插入
            node = Node(item)
            cur = self.__head
            count = 0
            while count < (pos - 1):
                count += 1
                cur = cur.next

            node.next = cur.next
            cur.next = node

    def search(self, item):
        """查找节点是否存在"""
        if self.is_empty():
            return False

        cur = self.__head

        while cur.next != self.__head:
            if cur.item == item:
                return True
            else:
                # 让游标继续执行
                cur = cur.next
        # 循环退出 cur指向尾节点
        if cur.item == item:
            return True

        return False

    def remove(self, item):
        """删除节点"""
        # 空链表
        if self.is_empty():
            return

        cur = self.__head
        pre = None

        # 头节点的元素就是要删除的元素
        if cur.item == item:
            # 链表中的节点不止一个
            if cur.next != self.__head:
                while cur.next != self.__head:
                    cur = cur.next
                # 循环结束 cur 指向尾结点

                cur.next = self.__head.next

                self.__head = cur.next
            else:
                self.__head = None
        # 第一个节点不是要删除的
        else:

            while cur.next != self.__head:
                if cur.item == item:
                    # 删除
                    pre.next = cur.next
                    return
                else:
                    # 游标继续往下走
                    pre = cur
                    cur = cur.next

            if cur.item == item:
                pre.next = cur.next


s = SinCycLinkList()

"""测试结果"""
print(s.is_empty())     # 判断是否为空链表
print(s.length())       # 判断链表长度

s.add(1)    # 在头部添加数据
s.add(2)
s.add(3)
s.add(4)
s.append(6)     # 在尾部添加数据
s.append(8)
s.travel()      # 遍历单向循环链表

s.insert(2, 7)  # 在指定位置插入数据
s.insert(2, 7)
s.travel()      # 遍历单向循环链表

s.remove(6)     # 删除数据
s.remove(7)
s.travel()      # 再次遍历

双向链表

class Node(object):
    """节点"""
    def __init__(self, item):
        # item存放数据元素
        self.item = item
        # next是下一个节点的标识
        self.next = None
        # perv是上一个节点的表示
        self.prev = None


class DoubleLinkList(object):
    """双向链表"""
    def __init__(self, node=None):
        self.__head = node

    def is_empty(self):
        """判断链表是否为空"""
        if self.__head == None:
            return True
        else:
            return False

    def length(self):
        """返回链表的长度"""
        # 空链表的情况下
        if self.is_empty():
            return 0

        # cur游标 指向首节点 用来遍历
        cur = self.__head
        # is 对象   == 数值是否相等
        count = 0
        while cur != None:
            count += 1
            # 将游标后移动一位
            cur = cur.next
        return count

    def travel(self):
        """遍历"""
        # 空链表
        if self.is_empty():
            return None

        cur = self.__head
        while cur != None:
            # 节点中的数据
            print(cur.item, end=' ')
            cur = cur.next
        print("")
        # 退出循环的时候,cur指向None

    def add(self, item):
        """链表头部添加元素"""
        # item 你要具体插入的数据
        node = Node(item)

        if self.is_empty():
            self.__head = node
        else:
            # 将node的next指向__head的头节点
            node.next = self.__head
            # 将__head的头节点的prev指向node
            self.__head.prev = node
            # 将__head指向node
            self.__head = node

    def append(self, item):
        """链表尾部添加元素"""
        # item 你要具体插入的数据
        node = Node(item)

        # 考虑空链表
        if self.is_empty():
            self.__head = node
        else:
            cur = self.__head  # None  node
            while cur.next != None:
                # 找到最后一个节点
                cur = cur.next
            # 循环结束之后,cur指向尾节点
            cur.next = node
            node.prev = cur

    def insert(self, pos, item):        # insert(2,100)  insert(-1,100)  insert(10,100)
        """指定位置添加元素"""
        # item 你要具体插入的数据
        # 头部插入
        if pos <= 0:
            self.add(item)
        # 尾部插入
        elif pos > self.length()-1:
            self.append(item)
        else:
            # 指定位置插入
            node = Node(item)
            cur = self.__head
            count = 0
            while count < (pos - 1):
                count += 1
                cur = cur.next
            # 循环结束,找到插入位置的前一个结点

            # 将node的prev指向cur
            node.prev =cur
            # 将node的next指向cur的下一个结点
            node.next = cur.next
            # 将cur的下一个结点的prev指向node
            cur.next.prev = node
            # 将cur的next指向node
            cur.next = node

    def search(self, item):
        """查找节点是否存在"""
        cur = self.__head
        while cur != None:
            if cur.item == item:
                return True
            else:
                # 让游标继续执行
                cur = cur.next
        return False

    def remove(self, item):
        """删除节点"""
        # remove(100)   [100,200,100] 删除一个100

        if self.is_empty():
            return
        else:
            cur = self.__head  # 空链表 None
            # 如果首结点的元素就是要删除的元素
            if cur.item == item:
                print(1)
                # 如果链表只有一个结点
                if cur.next == None:
                    self.__head = None
                else:
                    cur.next.prev = None
                    self.__head = cur.next
                return

            while cur != None:
                if cur.item == item:
                    print(2)
                    cur.prev.next = cur.next
                    if cur.next:
                        cur.next.prev = cur.prev
                    break
                cur = cur.next


d = DoubleLinkList()

"""测试结果"""
print(d.is_empty())
print(d.length())
d.add(1)
d.add(2)
d.add(3)
d.append(4)
d.append(10)
d.travel()

d.insert(-1, 0)
d.insert(-5, 0)
d.travel()

栈

class Stack(object):
    """用列表的方式来实现栈"""
    def __init__(self):
        self.__items = []

    def is_empty(self):
        """判断栈是否为空"""
        if self.__items == []:
            return True
        else:
            return False
        # return  self.__items == []

    def push(self, item):
        """添加一个新的元素item到栈顶"""
        self.__items.append(item)
        # self.__items.insert(0, item)

    def pop(self):
        """弹出栈顶元素"""
        return self.__items.pop()

    def peek(self):
        """返回栈顶元素"""
        if self.is_empty():
            return None
        else:
            return self.__items[-1]
            # return self.__items[len(self.__items)-1]

    def size(self):
        """返回栈的元素个数"""
        return len(self.__items)

    def travel(self):
        """遍历"""
        print(self.__items)
        # for item in self.__items:
        #     print(item)


s = Stack()

"""测试结果"""
print(s.is_empty())
s.push(1)
s.push(2)
s.push(3)
s.pop()     # 弹出栈顶item 3
s.travel()

print(s.size())
print(s.is_empty())
print(s.peek())     # 返回栈顶item而不弹出
s.travel()

队列

class Queue(object):
    def __init__(self):
        self.__items = []

    def enqueue(self, item):
        self.__items.append(item)

    def dequeue(self):
        """
        从队列头部删除一个元素
        :return: 队列头元素
        """
        return self.__items.pop(0)

    def is_empty(self):
        """
        判断一个队列是否为空
        :return: True or False
        """
        return self.__items == []

    def size(self):
        return len(self.__items)

    def travel(self):
        """遍历"""
        # print(self.__items)
        for item in self.__items:
            print(item, end=' ')
        print('')


q = Queue()

"""测试结果"""
print(q.is_empty())
q.enqueue(1)
q.enqueue(2)
q.enqueue(3)
q.enqueue(4)
q.travel()

q.dequeue()     # 队首item出队
q.travel()      # 再次遍历

冒泡排序

def bu_sort(lis):
    n = len(lis)
    for j in range(n-1):
        for i in range(0, n-1-j):
            if lis[i] > lis[i + 1]:
                lis[i], lis[i + 1] = lis[i + 1], lis[i]

选择排序

"""
首先在未排序列中找到最小元素
放到已排序列的起始位置
然后从未排序列中继续寻找最小元素
添加到已排序列的末尾
以此类推，直到所有元素均排序完毕。
"""


def selection_sort(lis):
    n = len(lis)
    for i in range(n-1):
        # 记录最小值的位置
        min_index = i
        # 从min_index 位置到末尾找到最小值  找最小值的下标
        for j in range(min_index, n):
            if lis[j] < lis[min_index]:   # lis[1] < lis[0]  26 < 54 min_index = 1
                min_index = j

        lis[i], lis[min_index] = lis[min_index], lis[i]

插入排序

"""
首先将数组中的数据分为两个区间
已排序区间和未排序区间
初始已排序区间只有一个元素
就是数组的第一个元素
插入排序的核心思想是取未排序区间中的元素
在已排序区间中找到合适的插入位置将其插入
并保证已排序区间数据一直有序
重复这个过程，直到未排序区间中元素为空，算法结束
"""


def insert_sort(lis):
    n = len(lis)
    # 从第二个位置,即下标为1的开始向前插入
    for i in range(1, n):
        # 从第i个元素 向前比较 如果小于前一个元素 交换
        for j in range(i, 0, -1):
            if lis[j] < lis[j - 1]:
                lis[j], lis[j - 1] = lis[j - 1], lis[j]

希尔排序

def shell_sort(lis):
    n = len(lis)
    gap = n//2  # python2  python3 存在区别
    while gap > 0:
        for i in range(gap, n):
            while i >= gap and lis[i-gap] > lis[i]:
                lis[i-gap], lis[i] = lis[i], lis[i-gap]
                i -= gap
        gap = gap//2

快速排序

def quick_sort(lis, start, end):
    if start >= end:
        return

    mid = lis[start]
    low = start
    high = end

    # 循环退出 low = high
    while low < high:
        while low < high and lis[high] >= mid:
            high -= 1
        lis[low] = lis[high]
        while low < high and lis[low] < mid:
            low += 1
        lis[high] = lis[low]

    lis[low] = mid
    quick_sort(lis, start, low-1)
    quick_sort(lis, low+1, end)


lis = [9, 21, 43, 27, 67, 31, 49, 55, 20]
quick_sort(lis, 0, len(lis)-1)

归并排序

"""
将数组分解最小之后合并两个有序数组
基本思路是比较两个数组的最前面的数
谁小就先取谁，取了后相应的指针就往后移一位
然后再比较，直至一个数组为空，最后把另一个数组的剩余部分复制过来即可。
"""


def merge_sort(lis):
    if len(lis) <= 1:
        return lis

    # 二分分解
    mid_index = len(lis) // 2
    left_lis = merge_sort(lis[:mid_index])
    right_lis = merge_sort(lis[mid_index:])

    l_index = 0
    r_index = 0
    result = []

    while l_index < len(left_lis) and r_index < len(right_lis):
        if left_lis[l_index] < right_lis[r_index]:
            result.append(left_lis[l_index])
            l_index += 1
        else:
            result.append(right_lis[r_index])
            r_index += 1

    result += left_lis[l_index:]
    result += right_lis[r_index:]
    return result   # 返回排完序的列表


lis = [14, 26, 63, 17, 77, 31, 44, 55, 20]
s = merge_sort(lis)