学习笔记 - 数据结构：数组

• 数组插入改进策略

"""
代码只实现想表达的逻辑

数组插入的改进，实现插入的时间复杂度为O(1)
"""

# 在第k个位置上插入元素e
def insert(array, e, k):
    length = len(array)
    array.insert(length, array[k-1])  # 这里只是模拟将元素迁移到末尾的操作
    array[k-1] = e

if __name__ == "__main__":
    x = [1, 2, 4, 5]  # 用list假装成数组
    insert(x, 10, 2)
    print(x)  # [1, 10, 4, 5, 2]

• 数组删除改进策略

import numpy as np

"""
a表示正常的数据
b表示标记数组，1表示需要删除

现在模拟，当空间满了需要将标记删除的元素真实地删除掉
"""
if __name__ == "__main__":
    a = np.random.randint(0, 20, size=10)
    b = np.random.randint(0, 2, size=10)

    # 用一个变量step表示移动的步伐
    step = 0
    length = len(a)

    print("删除前,当前数组长度：", length)
    for i in range(length):
        print("(",a[i], ":", b[i],")", end=" ")

    for i in range(length):
        if b[i] == 1:
            step += 1
            b[i] = 0
        else:
            # 每个不被删除的元素只移动一次
            a[i-step] = a[i]

    length -= step
    print("删除后,当前数组长度：", length)
    for i in range(length):
        print("(",a[i], ":", b[i],")", end=" ")

删除前,当前数组长度： 10
( 1 : 0 ) ( 18 : 0 ) ( 15 : 1 ) ( 10 : 1 ) ( 17 : 0 ) ( 12 : 1 ) ( 3 : 0 ) ( 1 : 1 ) ( 14 : 0 ) ( 11 : 0 ) 
删除后,当前数组长度： 6
( 1 : 0 ) ( 18 : 0 ) ( 17 : 0 ) ( 3 : 0 ) ( 14 : 0 ) ( 11 : 0 )