数据结构与算法--数组

数组

1、数组的定义

数组是具有一定顺序关系的若干对象组成的集合，组成数组的对象称为数组元素。

例如：

• 向量对应一维数组
• 矩阵对应二维数组

数组名表示群体的共性，即具有同一种数据类型；下标表示个体的个性，即各自占有独立的单元。

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

def test_list():
    N = input("请输入一个正整数：")
    N = int(N)
    A = []

    for i in range(N+1):
        if i >0 and i%5 == 0:
            A.append(i)
        elif i >0 and i%7 == 0:
            A.append(i)
    print(A)

2、托普利茨矩阵问题

如果一个矩阵的每一方向由左上到右下的对角线上具有相同元素，那么这个矩阵是托普利茨矩阵。
给定一个M x N的矩阵，当且仅当它是托普利茨矩阵时返回True

def toplitz(matrix):
    m = np.size(matrix,0)
    n = np.size(matrix,1)
    false = 0
    for i in range(m-1):
        for j in range(n-1):
            if matrix[i][j] != matrix[i+1][j+1]:
                false+=1
    if false:
        print(False)
    else:
        print(True)

matrix = [
    [1,2,3,4],
    [5,1,2,3],
    [9,5,1,2],
]

leetcode运行结果：

3、三数之和

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

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

解题思路：

• 首先对数组进行排序，排序后固定一个数 $nums[i]$，再使用左右指向 $nums[i]$后面的两端，数字分别为$nums[L]$ 和 $nums[R]$，计算三个数的和$sum$判断是否满足为 0，满足则添加进结果集
• 如果 $nums[i]$大于 0，则三数之和必然无法等于 0，结束循环
• 如果$nums[i]$== $nums[i-1]$，则说明该数字重复，会导致结果重复，所以应该跳过
• 当 $sum==0$ 时，$nums[L]==nums[L+1]$则会导致结果重复，应该跳过，L++
• 当 $sum==0$ 时，$nums[R]==nums[R-1$]则会导致结果重复，应该跳过，R−−
• 时间复杂度：$O(n^2)$ n 为数组长度

def three_sum(nums):
    nums.sort()
    res = []
    n = len(nums)
    for i in range(n):
        if nums[i] > 0:    #最小的数大于0，和不可能为0，直接退出
            break
        if i > 0 and nums[i] == nums[i - 1]:
            continue
        left = i + 1
        right = n - 1
        while left < right:
            nums_sum = nums[i] + nums[left] + nums[right]
            if nums_sum == 0:
                res.append([nums[i],nums[left],nums[right]])
                while left < right and nums[left] == nums[left+1]:
                    left+=1
                    continue
                while left < right and nums[right-1] == nums[right]:
                    right-=1
                    continue
                left+=1
                right-=1
            elif nums_sum < 0:
                left += 1
            elif nums_sum > 0:
                right -= 1

    print(res)
    print(nums)

nums = [-1, 0, 1, 2, -1, -4]

three_sum(nums)