剑指offer--algorithm3

2018-04-26  本文已影响0人  strive鱼

归零后才能重新出发,不留遗憾才是最好的故事。

题 5 --重建二叉树

关于树的考察,书中有这样的描述,一般指的为二叉树--每一个节点都有两个子节点,而考察二叉树,一般考察的为遍历 。关于树的题目多变,但是一般解决方法都是通过前序遍历、中序遍历、后序遍历等遍历思路,而每一种思路都有递归跟循环两种方法。
image.png
下开始解题
关于重建二叉树的思路,书中主要是根据前序遍历和中序遍历的两种遍历方式进行分析,分析的相当到位,只能盗图
image.png
image.png

程序代码如下:

class treenode(object):#定义一个树的节点
    def __init__(self,a):
        self.value=a
        self.left=None#定义左节点
        self.right=None#定义右节点

class solution(object):
    def reconstructiontree(self,pre,med):
        if not pre and not med:#如果为空,则返回None 
            return None 
        if set(pre)!=set(med):#如果两者不相等则说返回None
            return None 
        i=med.index(pre[0])#确定根节点的位置
        root=treenode(pre[0])#建立树
        root.left=self.reconstructiontree(pre[1:i+1],med[:i])#左侧树
        root.right=self.reconstructiontree(pre[i+1:],med[i+1:])#右侧数
        return root
    
def main():
    pre=[1,2,4,7,3,5,6,8]
    med=[4,7,2,1,5,3,8,6]
    test=solution()
    root=test.reconstructiontree(pre,med)
    print (root)
     
if __name__=='__main__':
    main()

程序的输出结果为一个树类型,也可以根据后序遍历,来自行修改程序代码

题6--队列和栈

题干基本意思就是用两个栈实现队列的功能---即利用先进后出的两个数据结构实现一个先进先出的数据结构

关于解题思路的逻辑,书中介绍的很清楚 mark


image.png image.png
image.png

下面为程序代码:

"""
栈:先进后出,要出的元素一定在栈顶,其实列表的pop方法就是此原理
队列:先进的先出
"""
class queue(object):
    def __init__(self):
        self.stack1=[]
        self.stack2=[]#声明了两个栈
    def push(self,a):#加入元素
        self.stack1.append(a)#利用append加入元素
        #print (self.stack1)
    def pop(self):
        if len(self.stack1)==0 and len(self.stack2)==0:
            return None
        elif len(self.stack2)==0:
            while len(self.stack1)>0:
                self.stack2.append(self.stack1.pop())#栈2接受栈1的弹出元素,这时候在栈1中先进的元素,在栈2中变成了后进元素,即弹出时顺序优先
        return self.stack2.pop()


def main():
    q=queue()#实例化
    q.push(1)
    q.push(2)
    q.push(3)
    #print (q.stack1)
    print (q.pop())
    print (q.pop())
    print (q.pop())
if __name__=='__main__':
    main()
上一篇下一篇

猜你喜欢

热点阅读