栈的实现

# -*- coding:utf-8 -*-
"""
数据结构的核心是,一个结构只有特定的操作, 可以实现特定的功能
栈的特点是 先进先出 , 一般有这几个操作
push 将一个元素存入栈中
pop 将一个元素从栈中取出, 并在栈中删除他

top 将一个元素从栈中取出
is_empty 查看栈是否为空
"""


# Node 类是一个节点, 有两个属性, 一个储存元素,一个储存指向另一个节点的引用
class Node(object):
    def __init__(self, element=None, next=None):
        self.element = element
        self.next = next

    # 这个函数会在print的时候自动调用, 就是把Node显示出来
    def __repr__(self):
        return str(self.element)


class Stack(object):
    # 初始化函数, 自动被调用
    # 初始化Stack()类的时候, 他有一个head的属性, 值是一个空的Node
    def __init__(self):
        self.head = Node()

    # 如果head的next的属性为空, 说明栈为空
    def is_empty(self):
        return self.head.next is None

    # 这样不行, 最后找不到head
    # # 创建一个node,
    # def push(self,element):
    #     n = Node(element)
    #     self.head.next = n
    #     self.head = n

    # 创建一个node, 并让他指向当前的head.next指向的元素, 在把head.next指向他
    def push(self, element):
        n = Node(element, self.head.next)
        self.head.next = n

    def pop(self):
        n = self.head.next
        if not self.is_empty():
            self.head.next = n.next
        return n


def test():
    s = Stack()
    s.push(1)

    print(s.pop())
    s.push(2)
    s.push(3)
    s.push(4)
    s.push(5)
    print(s.pop())
    print(s.pop())
    print(s.pop())
    print(s.pop())


if __name__ == "__main__":
    test()