剑指offer【50~59】

echobingo

发布于 2019-12-20 11:59:16

3520

发布于 2019-12-20 11:59:16

文章被收录于专栏：Bingo的深度学习杂货店Bingo的深度学习杂货店

题目链接：

剑指offer 50-59

Python 实现：

50. 第一个只出现一次的字符位置

使用大小为 256 的数组记录每个字符出现的次数。遍历两遍即可。

# -*- coding:utf-8 -*-
class Solution:
    def FirstNotRepeatingChar(self, s):
        # write code here
        dic = [0] * 256
        for i in range(len(s)):
            dic[ord(s[i])] += 1
        for i in range(len(s)):
            if dic[ord(s[i])] == 1:
                return i
        return -1

51. 数组中的逆序对

常规方法 O(n^2)，先 pass。可以使用归并排序的思想，在归并的过程中统计逆序对的数量。
比如在归并过程中，left = 1,3,5,7,9，right = 2,4,6,8,10，发现 3 > 2，则 left1: 的所有数都比 right0 大，就累加逆序对数量，后面的也同理。
详情可以参考博客剑指Offer（三十五）：数组中的逆序对
这样时间复杂度为 O(nlogn)，空间复杂度为 O(n)。但是 Python 卡了一下时间，AC 了 75%。

# -*- coding:utf-8 -*-
class Solution:
    def InversePairs(self, data):
        # write code here
        self.ans = 0  # 逆序对数量
        self.mergeSort(data)   # 归并排序
        return self.ans % 1000000007
    
    def mergeSort(self, nums):
        # 递归过程
        if len(nums) <= 1:
            return nums
        mid = len(nums) // 2
        left = self.mergeSort(nums[:mid])
        right = self.mergeSort(nums[mid:])
        return self.merge(left, right)

    # 归并过程 + 统计逆序对数量
    def merge(self, left, right):
        result = []  # 保存归并后的结果
        i = j = 0
        left_len = len(left)
        while i < len(left) and j < len(right):
            if left[i] <= right[j]:
                result.append(left[i])
                i += 1
            else:  # left[i] > right[j]
                self.ans += (left_len - i)  # 核心：说明 nums[i:] 都大于 nums[j]
                result.append(right[j])
                j += 1
        result = result + left[i:] + right[j:] # 剩余的元素直接添加到末尾
        return result

52. 两个链表的第一个公共结点

设 A 的长度为 a + c，B 的长度为 b + c，其中 c 为尾部公共部分长度，可知 a + c + b = b + c + a。
当访问链表 A 的指针 L1 访问到链表尾部时，令它从链表 B 的头部重新开始访问链表 B；同样地，当访问链表 B 的指针 L2 访问到链表尾部时，令它从链表 A 的头部重新开始访问链表 A。这样就能控制 L1 和 L2 能同时访问到公共结点。即 L1 总共走了 (a + c) + b，L2 总共走了 (b + c) + a。
这样，链表 A 和链表 B 都走了 a + b + c 步，时间复杂度为 O(n)，空间复杂度为 O(1)。

# -*- coding:utf-8 -*-
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None
class Solution:
    def FindFirstCommonNode(self, pHead1, pHead2):
        # write code here
        l1, l2 = pHead1, pHead2
        while l1 != l2:
            l1 = l1.next if l1 != None else pHead2
            l2 = l2.next if l2 != None else pHead1
        return l1 # 或者 l2

53. 数字在排序数组中出现的次数

排序数组，很明显二分查找，找到第一个 >= k 的元素索引以及第一个 > k 的元素索引，两者相减即为答案，即 lowerBound - upperBound**。时间复杂度为 O(logn)，空间复杂度为 O(1)。**

# -*- coding:utf-8 -*-
class Solution:
    def GetNumberOfK(self, data, k):
        # write code here
        if not data:
            return 0
        lo, hi = 0, len(data) - 1
        while lo < hi:
            mid = lo + (hi - lo) // 2
            if data[mid] >= k:
                hi = mid
            elif data[mid] < k:
                lo = mid + 1
        ind1 = lo if data[lo] >= k else len(data)  # >=k 的第一个元素
        lo, hi = 0, len(data) - 1
        while lo < hi:
            mid = lo + (hi - lo) // 2
            if data[mid] > k:
                hi = mid
            elif data[mid] <= k:
                lo = mid + 1
        ind2 = lo if data[lo] > k else len(data)  # # >k 的第一个元素
        return ind2 - ind1  # 相减即为答案

更简洁的，可以使用 Python 的 bisect 模块中的函数实现。可以参考博客：二分查找及其变形与Python的bisect模块的关系。

# -*- coding:utf-8 -*-
import bisect
class Solution:
    def GetNumberOfK(self, data, k):
        # write code here
        return bisect.bisect_right(data, k) - bisect.bisect_left(data, k)

54. 二叉查找树的第 K 个结点

中序遍历，找到第 k 个数即可。

#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    # 返回对应节点TreeNode
    def __init__(self):
        self.cnt = 0
        
    def KthNode(self, pRoot, k):
        # write code here
        if not pRoot or self.cnt >= k:  # self.cnt >= k 剪枝
            return None
        left = self.KthNode(pRoot.left, k)
        if left:
            return left
        self.cnt += 1
        if self.cnt == k:
            return pRoot
        right = self.KthNode(pRoot.right, k)
        if right:
            return right
        return None

55.1 二叉树的深度

递归左右子树找最大高度。注意：根的高度为 1。

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def TreeDepth(self, pRoot):
        # write code here
        if not pRoot:
            return 0
        return max(self.TreeDepth(pRoot.left), self.TreeDepth(pRoot.right)) + 1

55.2 平衡二叉树

平衡二叉树（AVL）的左右子树的高度差不超过 1，因此引入求高度的函数。并且，判断该树的每个结点的左右子树是否也满足 AVL 的定义。

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def IsBalanced_Solution(self, pRoot):
        # write code here
        if not pRoot:
            return True
        if abs(self.TreeDepth(pRoot.left) - self.TreeDepth(pRoot.right)) > 1:  # 不满足 AVL 的定义
            return False
        return self.IsBalanced_Solution(pRoot.left) and self.IsBalanced_Solution(pRoot.right)
        
    def TreeDepth(self, pRoot):
        if not pRoot:
            return 0
        return max(self.TreeDepth(pRoot.left), self.TreeDepth(pRoot.right)) + 1

改进：自顶向下在调用求树的高度的函数时，有很多重复的操作。因此，可以使用自底向上的方法，一边计算树的高度，一边判断是否是 AVL 树。

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def IsBalanced_Solution(self, pRoot):
        # write code here
        self.isBalance = True
        self.TreeDepth(pRoot)  # 在求树的高度过程中判断 AVL
        return self.isBalance
        
    def TreeDepth(self, pRoot):
        if not pRoot:
            return 0
        left = self.TreeDepth(pRoot.left)
        right = self.TreeDepth(pRoot.right)
        if abs(left - right) > 1:   # 不满足 AVL 定义，修改 self.isBalance 的值
            self.isBalance = False
        return max(left, right) + 1

56. 数组中只出现一次的数字

如果只出现一次的数字只有一个，很好做，就是全部异或即可。但是，只出现一次的数字有两个怎么做呢？

假设只出现一次的数字为 x 和 y，首先，还是先全部异或得到一个结果 xor，则 x ^ y = xor（相同的数字异或后抵消为 0）
因为 x 和 y 肯定不同，那么它们的二进制表示中肯定有一位一个是 0，一个是 1。比如 x = 6 (110)，y = 4 (100)，xor = 2 (10)，则我们对 xor 从后往前找到倒数第一个 1 的位置 bits（倒数第 2 位），则以这个 1 为界限，x 和 y 的倒数第 2 位肯定是不同。
因此，对原来的数组重新异或，将 bits 位置为 1 的全部异或，bits 为 0 的全部异或，就是最终的两个返回的结果。

# -*- coding:utf-8 -*-
class Solution:
    # 返回[a,b] 其中ab是出现一次的两个数字
    def FindNumsAppearOnce(self, array):
        # write code here
        xor = 0
        ans = [0, 0]
        for num in array:
            xor ^= num
        bits = 0  # 找到倒数第 i 位为 1
        while xor & 1 << bits == 0:
            bits += 1
        for num in array:
            if num >> bits & 1 == 1:  # num 的倒数第 i 位为 1
                ans[0] ^= num
            else:  # num 的倒数第 i 位为 0
                ans[1] ^= num
        return ans

57.1 和为 S 的两个数字

双指针指向首尾，往中间走，碰到第一对和为 S 的就是答案。

# -*- coding:utf-8 -*-
class Solution:
    def FindNumbersWithSum(self, array, tsum):
        # write code here
        lo, hi = 0, len(array) - 1
        while lo < hi:   # 不能指向相同的数字
            if array[lo] + array[hi] == tsum:
                return [array[lo], array[hi]]
            elif array[lo] + array[hi] < tsum:
                lo += 1
            else:
                hi -= 1
        return []

57.2 和为 S 的连续正数序列

经典的滑动窗口例题，只不过该题的窗口大小不确定。用一个遍历 left 记录窗口左侧的值，window_sum 将窗口中数进行累加。当发现 window_sum >= S 时，相等时输出结果，更新 left 和 window_sum。

# -*- coding:utf-8 -*-
class Solution:
    def FindContinuousSequence(self, tsum):
        # write code here
        if tsum <= 1:  # 特殊情况
            return []
        ans = []
        left, window_sum = 1, 0  # left 记录窗口左界
        for i in range(1, (tsum + 1) // 2 + 1):  # 窗口中至少两个数
            window_sum += i  # 窗口中的累加和
            while window_sum >= tsum:
                if window_sum == tsum:  # 输出一组结果
                    ans.append([j for j in range(left, i + 1)])
                window_sum -= left  # 修改窗口中的累加和
                left += 1  # 修改窗口的左界
        return ans

58.1 翻转单词顺序列

如果要求空间复杂度为 O(1)，即只能使用字符串本身，该怎么操作呢？

如 s = "I am a student."，先将各个单词翻转，得 s = ".tneduts a ma I"；
再将整个字符串翻转，得 s = "student. a am I"

这样就可以在字符串本身上做修改，使得空间复杂度为 O(1) 了。

注意：但是使用 Python 实现得话，由于不能修改字符串本身，所以还是先要将字符串转化为列表。但是如果使用 C++ 的字符数组，就不用开辟空间了。

# -*- coding:utf-8 -*-
class Solution:
    def ReverseSentence(self, s):
        # write code here
        def reverseWord(s, i, j):
            while i < j:
                s[i], s[j] = s[j], s[i]
                i += 1
                j -= 1
        
        s = list(s)  # python 中 s 不能修改，先转化为 list 
        start = 0  # start 指向当前单词的起始位置
        lens = len(s)
        for i in range(lens + 1):
            if i == lens or s[i] == ' ':
                reverseWord(s, start, i - 1) # 先翻转每个单词
                start = i + 1  # 下一个单词的起始位置
        reverseWord(s, 0, lens - 1)  # 再将整个句子翻转
        return "".join(s)

58.2 左旋转字符串

如果也不能使用空间，怎么做呢？参考上面的 58.1，如 s = "abcXYZdef"，n = 3，先将 "abc" 和 "XYZdef" 分别翻转，得到 "cbafedZYX"，然后再把整个字符串翻转得到 "XYZdefabc"。这样就可以空间复杂度为 O(1) 了。

# -*- coding:utf-8 -*-
class Solution:
    def LeftRotateString(self, s, n):
        # write code here
        def reverseWord(s, i, j):
            while i < j:
                s[i], s[j] = s[j], s[i]
                i += 1
                j -= 1
        
        if not s:
            return ""
        s = list(s)
        lens = len(s)
        n %= lens  # 循环左移
        reverseWord(s, 0, n - 1)  # 先将前 n 个字符翻转
        reverseWord(s, n, lens - 1)  # 再将后面的字符翻转
        reverseWord(s, 0, lens - 1)  # 最后将整个字符串翻转
        return "".join(s)  # 再转化回字符串

59. 滑动窗口的最大值

使用双向递减队列，队列中始终维护的是窗口中的递减值。

如果队列的大小达到了 size，则应该把队列最前面的数字删除掉。
如果从最右边加入了一个较大的数字，需要从右开始退队列（while 循环），使得队列是单调递减的。
由于双向队列中，从左边出队列和从右边出队列的操作时间复杂度均为 O(1)，因此该算法的时间复杂度为 O(n)，空间复杂度为 O(n)。

例如，num = 2,3,4,2,6,2,5,1，size = 3，双向队列的变化情况为 dq： 2 -> 3 -> 4 -> 4,2 -> 6 -> 6,2 -> 6,5 -> 5,1。

# -*- coding:utf-8 -*-
import collections
class Solution:
    def maxInWindows(self, num, size):
        # write code here
        if size > len(num) or size < 1:
            return []
        dq = collections.deque()  # [num[i], i]
        ans = []
        for i in range(len(num)):
            if dq and i - dq[0][1] >= size:
                dq.popleft()  # O(1)
            while dq and num[i] > dq[-1][0]: # 从后往前删除比 num[i] 大的数
                dq.pop()  # O(1)
            dq.append([num[i], i])
            if i >= size - 1:
                ans.append(dq[0][0])  # 队列首部始终最大
        return ans

剑指 offer 终于过了一遍，大多数题目还是很简单的，但是题目具有代表性，涉及链表、数组、深搜回溯、字符串、数组、数学、位运算、动态规划等。

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

如有侵权请联系 cloudcommunity@tencent.com 删除

python

二叉树

编程算法

本文分享自作者个人站点/博客前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体同步曝光计划，欢迎热爱写作的你一起参与！

python

二叉树

编程算法

登录后参与评论

0 条评论

热度