1、二维数组中的查找
有一个二维数组(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
思路1:遍历数组
时间复杂度:O(m*n)
class Solution:
def Find(self, target, array): #array二维列表
raw = len(array) #行数
col = len(array[0]) #列数
for i in range(raw):
for j in range(col):
# 如果目标值等于数组中的值,则找到
if target == array[i][j]:
return True
return False
(错误思路:从左上角开始查找是不可行,因为以左上角为基准遍历时,要查找的数字可能分布在两侧,所以不能很好的缩小搜索范围。)
思路2:以左下角或右上角开始查找
时间复杂度:O(m+n) (行数+列数)
def Find2(self, target, array):
raw = len(array)
col = len(array[0])
#以左下角为起点开始找
i = raw-1
j = 0
while (i>=0) and (j<col):
if target>array[i][j]:
j += 1
elif target<array[i][j]:
i -= 1
else:
return True
return False
另:对折查找(以行列的中点开始判断大小)、十字分割(从数组整体的中点开始)
2、重建二叉树
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。
前序遍历:根→左→右
中序遍历:左→根→右
思路1:递归重建
在前序序列中,第一个元素A就是根节点;
此时在中序序列中,A的左边(B、C)就是A的左孩子及其子节点,在A的右边(D、E…I)就是A的右孩子及其子节点;
在前序序列中第二个是B,所以B就是A的左孩子;
而在中序序列,B的左边C是B的左孩子及其子节点,在B的右边(同是也在A的左边)的就是B的右孩子及其子节点…
以此类推.这就是利用递归来重建二叉树。
递归调用:将左子树和右子树分别再看成单独一颗树,将其前序遍历序列、中序遍历序列分别传入到该函数中,便可得到左子树的根结点、右子树的根结点。此时需要用第一步得到的根结点连接它们;
递归调用的终止条件:直到传入数组为空,说明已经没有节点,直接返回null。
class TreeNode: # 先定义树的基本结构
def __init__(self, x):
self.val=x
self.left=None
self.right=None
class Solution:
def reConstructBinaryTree(self, pre, tin):
if not pre or not tin: # 检测有无中序和先序遍历
return None
root = TreeNode(pre[0])
val = tin.index(pre[0]) # .index可以查找列表里面值的下表
#递归调用
root.left = self.reConstructBinaryTree(pre[1:val+1], tin[:val])
root.right = self.reConstructBinaryTree(pre[val+1:], tin[val+1:])
return root