python实现数组，利用动态数组解决数据存放问题，托普利茨矩阵，三数之和

以下代码都为 python 版本

1 利用动态数组解决数据存放问题

编写一段代码，要求输入一个整数N，用动态数组A来存放2~N之间所有5或7的倍数，输出该数组。

示例：

输入：
N = 100 

输出：
5 7 10 14 15 20 21 25 28 30 35 40 42 45 49 50 55 56 60 63 65 70 75 77 80 84 85 90 91 95 98 100

代码：

a = int(input())
b = [None if a < 5 else i for i in range(5, a+1, 5)]
c = [None if a < 7 else i for i in range(7, a+1, 7)]
d = list(set(b+c))
for i in d:
    print(i)

2 托普利茨矩阵问题

如果一个矩阵的每一方向由左上到右下的对角线上具有相同元素，那么这个矩阵是托普利茨矩阵。

给定一个M x N的矩阵，当且仅当它是托普利茨矩阵时返回True。

示例：

输入:
matrix = [
  [1,2,3,4],
  [5,1,2,3],
  [9,5,1,2]
]

输出: True

解释:

在上述矩阵中, 其对角线为: "[9]", "[5, 5]", "[1, 1, 1]", "[2, 2, 2]", "[3, 3]", "[4]"。 各条对角线上的所有元素均相同, 因此答案是True。

代码：

class Solution(object):
    def isToeplitzMatrix(matrix):
        for i in range(len(matrix) - 1):
            if not matrix[i][:-1] == matrix[i+1][1:]:
                return False
            return True

3 三数之和

给定一个包含 n 个整数的数组nums，判断nums中是否存在三个元素a，b，c，使得a + b + c = 0？找出所有满足条件且不重复的三元组。

注意：答案中不可以包含重复的三元组。

示例：

给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
  [-1, 0, 1],
  [-1, -1, 2]
]

代码：

"""
算法流程：
    特判，对于数组长度 n，如果数组为 null 或者数组长度小于 3，返回 []。
    对数组进行排序。
    遍历排序后数组：
        若 nums[i]>0：因为已经排序好，所以后面不可能有三个数加和等于 0，直接返回结果。
        对于重复元素：跳过，避免出现重复解
        令左指针 L=i+1，右指针 R=n−1，当 L<R时，执行循环：
            当nums[i]+nums[L]+nums[R]==0，执行循环，判断左界和右界是否和下一位置重复，去除重复解。并同时将 L,RL,RL,R 移到下一位置，寻找新的解
            若和大于0，说明 nums[R] 太大，R 左移
            若和小于0，说明 nums[L] 太小，L 右移
"""
class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        n = len(nums)
        res = []
        if(not nums or n < 3):
            return []
        nums.sort()
        for i in range(n):
            if (nums[i] > 0):
                return res
            if (i > 0 and nums[i] == nums[i-1]):
                continue
            left = i+1
            right = n-1
            while(left < right):
                if(nums[i] + nums[left] + nums[right] == 0):
                    res.append([nums[i], nums[left], nums[right]])
                    while(left < right and nums[left] == nums[left + 1]):
                        left += 1
                    while(left < right and nums[right] == nums[right - 1]):
                        right -= 1
                    left += 1
                    right -= 1
                elif(nums[i] + nums[left] + nums[right] > 0):
                    right -= 1
                else:
                    left += 1
        return res

三数之和：https://leetcode-cn.com/problems/3sum/

学习计划：https://github.com/datawhalechina/team-learning/blob/master/数据结构与算法（上）/学习任务.md

参考资料：王争《数据结构与算法之美》https://time.geekbang.org/column/article/40961

欢迎关注微信公众号 shinerise，与你一起慢慢进步~