数据结构——数组

数组是程序中最常见的数据结构,它可以存储一个固定大小的相同类型元素的顺序集合(强类型语言)。数组的元素都是由连续的内存位置组成。最低的地址对应第一个元素,最高的地址对应最后一个元素,通过索引可以非常容易找到某一个元素。

大多数时候我们需要使用一个大小可变的数组(C#、Python中的list),本文就基于数组来实现一个动态数组,由于在Python中的列表已经对数组封装的很好,这里我们使用C#来实现一个List。在后续介绍数据结构文章中,我会使用python和C#分别来实现相应的数据结构。

动态数组和普通数组在用户使用上没有区别,我们定义一个类MyArray,内部维护一个数组,并不需要实现太多方法,最核心的是提供扩容、索引和增删功能。

class MyArray<T>
    {
        private T[] array; //存储数组
        private int count; //存储数组大小
        private int capacity; //数组容量
        
        public int Count { get { return count; } }
        public int Capacity {  get { return capacity; } }

        //重载构造函数 接收一个初始容量
        public MyArray(int capacity){ }

        // 改变数组大小
        private void resize(int capacity) { }

        //在指定索引处插入元素
        public void Insert(int index,T item) { }

        //移除指定位置元素
        public void RemoveAt(int index)  { }

        //正序获取元素位置
        public int IndexOf(T item) { }

        //实现索引器
        public T this[int index]  { }
    }

下面我们按上面的方法一点一点来实现我们的MyArray

 1.实现构造函数

构造函数在功能上是为了初始化数组,所以用户可以传递一个初始数组容量,当然考虑到我们的数组本身是动态的,所以如果用户不传入初始容量时,我们应该使用默认大小创建数组。

public MyArray(int capacity)
        {
            //接收一个初始容量capacity
            if (capacity > 0)
            {
                this.capacity = capacity;
                array = new T[capacity];
            }
            else
             throw new Exception("列表容量必须大于0");
        }

       public MyArray()
        {
            //构造函数 初始化默认数组
            capacity = 4;
            array = new T[capacity];
        }

   在这里我们设置如果用户没有传入capacity,默认数组大小为4。我们还维护了一个变量this.capacity,其实这个变量并不必要,可以直接通过array.length获得数组容量。

2.实现数组扩容方法

思路十分简单,我们实例化一个新数组,把旧数组的数据复制到新数组即可。

private void resize(int capacity) {
    // 改变数组大小
    T[] newarray = new T[capacity];
    Array.Copy(array, newarray, count);
    array = newarray;
    this.capacity = capacity;
}

3.增加元素—Insert方法

把一个元素插入数组指定位置,可以分两种情况讨论:一是追加到数组末尾,二是插入到数组中。

第一种情况处理起来很简单,因为我们定义的类维护了一个count变量,它记录了数组的实际大小,所以我们只要赋值array[count],count++就可以实现功能。

第二种情况代表原来的位置已经有元素了,那么原位置之后的元素都应该集体向后挪一个位置,array[i+1] = array[i],我们可以用一个for循环来实现。

public void Insert(int index,T item)
{
    //在指定索引处插入元素
    if (index > count || index <0)
    {
        throw new Exception("索引超出范围");
    }
    if (capacity == count)
    {
        //数组扩容
        resize(capacity * 2);
    }
    if (index == count)
    {
        //第一种情况,在末尾追加元素追加元素
        array[count] = item;
        count++;
    } else
    {
        for (int i = count - 1; i>=index; i--)
        {
            //最后一次循环应为array[index+1] = array[index]
            array[i + 1] = array[i];
        }
        array[index] = item;
        count++;
    }
}

值得注意的是,插入元素可能会超出数组大小,所以我们做了一层capacity==count的判断,如果为真,我们就调用resize方法,将数组扩容至原来的两倍。

4.删除指定位置元素

删除元素的思路和增加元素的思路相反,把索引为i的元素删除后,后面的元素应该前进一位。

public void RemoveAt(int index)
{
    //移除指定位置元素
    if (index >= count || index < 0)
    {
        throw new Exception("索引超出范围");
    } else
    {
        for (int i = index; i < count - 1; i++)
        {
            //只要实现array[index] = array[index+1]
            //若i=count-1;则i+1可能会出现超出索引的情况,故条件为i<count-1
            array[i] = array[i + 1];
        }
        count--;
        if (count < capacity / 4) {
            resize(capacity/2);
        }
    }
}

在删除元素中,我们最后调用了resize方法,当元素个数小于数组的1/4时,我们把数组缩小至原来的1/2。

5.查找元素的索引

十分简单,循环数组即可

public int IndexOf(T item)
{
    //正序获取元素位置 若无返回-1
    for (int i = 0; i < count; i++)
    {
        if (array[i].Equals(item))
        {
            return i;
        }
    }
    return -1;
}

6.实现索引器

public T this[int index]
{
    get { return array[index]; }
    set { array[index] = value; }
} 

好了,到现在我们的MyArray最核心的功能完成了,当然你可以为它添加其他方法,让它在用户使用体验上,和原生数组更为相近。最后我们来看看各项操作的时间复杂度。

Insert方法,在末尾添加元素时间复杂度为O(1),在数组最前面添加元素为O(n),均摊时间复杂度为O(n)

RemoveAt方法,在末尾删除元素时间复杂度为O(1),在数组最前面添加元素为O(n),均摊时间复杂度为O(n)

IndexOf和resize方法,遍历数组,时间复杂度为O(n)

索引器,按索引访问元素,时间复杂度为O(1)

相关推荐