python-剑指offer16-20

16、树

操作给定的二叉树,将其变换为源二叉树的镜像。

# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    # 返回镜像树的根节点
    def Mirror(self, root):
        # write code here
        if root == None:
            return None
        if root.left != None or root.right != None:
            root.left,root.right = root.right,root.left
            self.Mirror(root.left)
            self.Mirror(root.right)

17、数组

输入一个矩阵,按照从外向里以顺时针的顺序依次打印出每一个数字,例如,如果输入如下4 X 4矩阵: 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 则依次打印出数字1,2,3,4,8,12,16,15,14,13,9,5,6,7,11,10.

方法一:

# -*- coding:utf-8 -*-
class Solution:
    # matrix类型为二维列表,需要返回列表
    def printMatrix(self, matrix):
        # write code here
        res=[]
        i,j=0,0
        d=0
        vis=[[0 for _ in range(len(matrix[0]))] for _ in range(len(matrix))]
        tmp=[[0,1],[1,0],[0,-1],[-1,0]]
        while i>=0 and i<len(matrix) and j>=0 and j<len(matrix[0]) and vis[i][j]==0:
            res.append(matrix[i][j])
            vis[i][j]=1
            while i+tmp[d][0]>=0 and i+tmp[d][0]<len(matrix) and j+tmp[d][1]>=0 and j+tmp[d][1]<len(matrix[0]) and vis[i+tmp[d][0]][j+tmp[d][1]]==0:
                i+=tmp[d][0]
                j+=tmp[d][1]
                res.append(matrix[i][j])
                vis[i][j]=1
            d=(d+1)%4
            i+=tmp[d][0]
            j+=tmp[d][1]
        return res

方法二:

# -*- coding:utf-8 -*-
class Solution:
    # matrix类型为二维列表,需要返回列表
    def printMatrix(self, matrix):
        # write code here
        if not matrix:
            return matrix
        rows = len(matrix)
        cols = len(matrix[0])
        result = []
        start = 0
        # 判断是否继续往里走一圈
        while cols > start * 2 and rows > start * 2:
            # 走一圈的代码
            end_x = cols - 1 - start
            end_y = rows - 1 - start
            # 打印第一行
            for i in range(start, end_x+1):
                result.append(matrix[start][i])
            # 打印最后一列
            if start < end_y:
                for i in range(start+1, end_y+1):
                    result.append(matrix[i][end_x])
            # 打印最后一行
            if start < end_x and start < end_y:
                for i in range(end_x-1, start-1, -1):
                    result.append(matrix[end_y][i])
            # 打印第一列
            if start < end_x and start < end_y - 1:
                for i in range(end_y-1, start, -1):
                    result.append(matrix[i][start])
            start += 1
        return result

方法三:

# -*- coding:utf-8 -*-
class Solution:
    # matrix类型为二维列表,需要返回列表
    def printMatrix(self, matrix):
        if not matrix:
            return matrix
        result = []
        while len(matrix) > 0:
            result.extend(matrix[0])
            matrix = zip(*matrix[1:])[::-1]
        return result

18、栈

定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.s1=[]
        self.s2=[]
    def push(self, node):
        # write code here
        self.s1.append(node)
        if len(self.s2)==0:
            self.s2.append(node)
        else:
            if node>self.s2[-1]:
                self.s2.append(self.s2[-1])
            else:
                self.s2.append(node)
    def pop(self):
        # write code here
        if len(self.s1)==0:
            return None
        del self.s2[-1]
        res=self.s1[-1]
        del self.s1[-1]
        return res
    def top(self):
        # write code here
        if len(self.s1)==0:
            return None
        else:
            return self.s1[-1]
    def min(self):
        # write code here
        if len(self.s1)==0:
            return None
        else:
            return self.s2[-1]

较好的思路:开一个新栈,此栈与数值栈一一对应,在元素入栈时,就记录处于每一个位置时的当前最小值,这样不论什么情况,不论之前如何出栈入栈,都能立即获取到最小值。

python-剑指offer16-20

19、栈

输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否可能为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。(注意:这两个序列的长度是相等的)

class Solution:
    def IsPopOrder(self, pushV, popV):
        # write code here
        stack = []
        cur = 0
        for pop in popV:
            while len(stack)==0 or pop != stack[-1] and cur < len(pushV):
                stack.append(pushV[cur])
                cur +=1
            if cur == len(pushV) and stack[-1] != pop:
                return False
            else:
                stack.pop()
        return True

解题思路:例如对于压入顺序 12345和弹出顺序45312,首先读到4,则从压入序列里依次压入1234,然后弹出4;读到5,则压入5,弹出5;读到3,栈顶恰好为3,弹出3;读到1,栈顶为2,应当继续压入直到压入1,但压入序列已经读完,没找到1,所以错误。

20、树

从上往下打印出二叉树的每个节点,同层节点从左至右打印。

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    # 返回从上到下每个节点值列表,例:[1,2,3]
    def PrintFromTopToBottom(self, root):
        # write code here
        if not root:
            return []
        stack=[root]
        res=[]
        while stack:
            for i in range(len(stack)):
                t=stack.pop(0)
                res.append(t.val)
                if t.left:
                    stack.append(t.left)
                if t.right:
                    stack.append(t.right)
        return res

解题思路:树的层次遍历,注意的是当root为空时要返回[],而不是None。

相关推荐