剑指offer4~6题
4、重建二叉树
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
前序遍历:首先访问根结点然后遍历左子树,最后遍历右子树。
中序遍历:首先遍历左子树,然后访问根结点,最后遍历右子树.
思路:
根据前序遍历获得根节点的值,然后在中序遍历中获取根节点的位置。中序遍历的根节点左边为左子树,右边为右子树。继而对左子树和右子树继续同样的操作。
# -*- coding:utf-8 -*-
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
# 返回构造的TreeNode根节点
def reConstructBinaryTree(self, pre, tin):
# write code here
if not pre or not tin:
return None
root = TreeNode(pre.pop(0))
index = tin.index(root.val)
root.left = self.reConstructBinaryTree(pre, tin[:index])
root.right = self.reConstructBinaryTree(pre, tin[index + 1:])
return root
5、用两个栈实现队列
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
思路:
栈是后进先出,队列是先进先出。可以将数据push进栈1,再push进栈2,最终由栈2pop出来。
class Solution:
def __init__(self):
self._stack1 = []
self._stack2 = []
def push(self, node):
# write code here
self._stack1.append(node)
def pop(self):
# return xx
if self._stack2:
return self._stack2.pop()
elif not self._stack1:
return None
else:
while self._stack1:
self._stack2.append(self._stack1.pop())
return self._stack2.pop()
6、旋转数组的最小数字
把一个数组最开始的若干个元素搬到数组的末尾,我们称之为数组的旋转。 输入一个非减排序的数组的一个旋转,输出旋转数组的最小元素。 例如数组{3,4,5,1,2}为{1,2,3,4,5}的一个旋转,该数组的最小值为1。 NOTE:给出的所有元素都大于0,若数组大小为0,请返回0。
思路:
旋转数组左边的最小值大于右边最大值。可以取一个中间值用来判断最小位置。
class Solution:
def minNumberInRotateArray(self, rotateArray):
# write code here
if rotateArray == []:
return 0
_len = len(rotateArray)
left = 0
right = _len - 1
#二分查找
while left <= right:
#除以2,即找到中间位置
mid = int((left + right) / 2)
if rotateArray[mid]<rotateArray[mid-1]:
return rotateArray[mid]
#注意有“=”这种情况
if rotateArray[mid] >= rotateArray[right]:
#说明在[mid,right]之间
left = mid + 1
else:
#说明在[left,mid]之间
right = mid - 1
return rotateArray[mid]