环形数组技巧

简介: 环形数组通过模运算在逻辑上将线性数组首尾相连,利用start和end指针实现头部O(1)增删。虽物理上非环形,但通过取余操作让指针循环移动,结合左闭右开区间设计,高效支持动态扩容缩容,适用于双端队列等场景。

😸一句话总结
环形数组技巧利用求模(余数)运算,将普通数组变成逻辑上的环形数组,可以让我们用 O(1) 的时间在数组头部增删元素。
环形数组原理
数组可能是环形的么?不可能。数组就是一块线性连续的内存空间,怎么可能有环的概念?
但是,我们可以在「逻辑上」把数组变成环形的,比如下面这段代码:
// 长度为 5 的数组
int[] arr = new int[]{1, 2, 3, 4, 5};
int i = 0;
// 模拟环形数组,这个循环永远不会结束
while (i < arr.length) {
System.out.println(arr[i]);
i = (i + 1) % arr.length;
}
这段代码的关键在于求模运算 %,也就是求余数。当 i 到达数组末尾元素时,i + 1 和 arr.length 取余数又会变成 0,即会回到数组头部,这样就在逻辑上形成了一个环形数组,永远遍历不完。
这就是环形数组技巧。这个技巧如何帮助我们在 O(1) 的时间在数组头部增删元素呢?
是这样,假设我们现在有一个长度为 6 的数组,现在其中只装了 3 个元素,如下(未装元素的位置用 标识):
[1, 2, 3,
, , ]

现在我们要在数组头部删除元素 1,那么我们可以把数组变成这样:
[, 2, 3, , , ]
即,我们仅仅把元素 1 的位置标记为空,但并不做数据搬移。
此时,如果我们要在数组头部增加元素 4 和元素 5,我们可以把数组变成这样
:[4, 2, 3, , , 5]

你可以看到,当头部没有位置添加新元素时,它转了一圈,把新元素加到尾部了。

核心原理
上面只是让大家对环形数组有一个直观地印象,环形数组的关键在于,它维护了两个指针 start 和 end,start 指向第一个有效元素的索引,end 指向最后一个有效元素的下一个位置索引。
这样,当我们在数组头部添加或删除元素时,只需要移动 start 索引,而在数组尾部添加或删除元素时,只需要移动 end 索引。
当 start, end 移动超出数组边界(< 0 或 >= arr.length)时,我们可以通过求模运算 % 让它们转一圈到数组头部或尾部继续工作,这样就实现了环形数组的效果。
动手环节
关键点、注意开闭区间
在我的代码中,环形数组的区间被定义为左闭右开的,即 [start, end) 区间包含数组元素。所以其他的方法都是以左闭右开区间为基础实现的。
那么肯定就会有读者问,为啥要左闭右开,我就是想两端都开,或者两端都闭,不行么?
理论上,你可以随意设计区间的开闭,但一般设计为左闭右开区间是最方便处理的。
因为这样初始化 start = end = 0 时,区间 [0, 0) 中没有元素,但只要让 end 向右移动(扩大)一位,区间 [0, 1) 就包含一个元素 0 了。
如果你设置为两端都开的区间,那么让 end 向右移动一位后开区间 (0, 1) 仍然没有元素;如果你设置为两端都闭的区间,那么初始区间 [0, 0] 就已经包含了一个元素。这两种情况都会给边界处理带来不必要的麻烦,如果你非要使用的话,需要在代码中做一些特殊处理。
最后,我给出一个支持泛型的 Java 实现,你可以参考一下:
public class CycleArray {
private T[] arr;
private int start;
private int end;
private int count;
private int size;

public CycleArray() {
    this(1);
}

@SuppressWarnings("unchecked")
public CycleArray(int size) {
    this.size = size;
    // 因为 Java 不支持直接创建泛型数组,所以这里使用了类型转换
    this.arr = (T[]) new Object[size];
    // start 指向第一个有效元素的索引,闭区间
    this.start = 0;
    // 切记 end 是一个开区间,
    // 即 end 指向最后一个有效元素的下一个位置索引
    this.end = 0;
    this.count = 0;
}

// 自动扩缩容辅助函数
@SuppressWarnings("unchecked")
private void resize(int newSize) {
    // 创建新的数组
    T[] newArr = (T[]) new Object[newSize];
    // 将旧数组的元素复制到新数组中
    for (int i = 0; i < count; i++) {
        newArr[i] = arr[(start + i) % size];
    }
    arr = newArr;
    // 重置 start 和 end 指针
    start = 0;
    end = count;
    size = newSize;
}

// 在数组头部添加元素,时间复杂度 O(1)
public void addFirst(T val) {
    // 当数组满时,扩容为原来的两倍
    if (isFull()) {
        resize(size * 2);
    }
    // 因为 start 是闭区间,所以先左移,再赋值
    start = (start - 1 + size) % size;
    arr[start] = val;
    count++;
}

// 删除数组头部元素,时间复杂度 O(1)
public void removeFirst() {
    if (isEmpty()) {
        throw new IllegalStateException("Array is empty");
    }
    // 因为 start 是闭区间,所以先赋值,再右移
    arr[start] = null;
    start = (start + 1) % size;
    count--;
    // 如果数组元素数量减少到原大小的四分之一,则减小数组大小为一半
    if (count > 0 && count == size / 4) {
        resize(size / 2);
    }
}

// 在数组尾部添加元素,时间复杂度 O(1)
public void addLast(T val) {
    if (isFull()) {
        resize(size * 2);
    }
    // 因为 end 是开区间,所以是先赋值,再右移
    arr[end] = val;
    end = (end + 1) % size;
    count++;
}

// 删除数组尾部元素,时间复杂度 O(1)
public void removeLast() {
    if (isEmpty()) {
        throw new IllegalStateException("Array is empty");
    }
    // 因为 end 是开区间,所以先左移,再赋值
    end = (end - 1 + size) % size;
    arr[end] = null;
    count--;
    // 缩容
    if (count > 0 && count == size / 4) {
        resize(size / 2);
    }
}

// 获取数组头部元素,时间复杂度 O(1)
public T getFirst() {
    if (isEmpty()) {
        throw new IllegalStateException("Array is empty");
    }
    return arr[start];
}

// 获取数组尾部元素,时间复杂度 O(1)
public T getLast() {
    if (isEmpty()) {
        throw new IllegalStateException("Array is empty");
    }
    // end 是开区间,指向的是下一个元素的位置,所以要减 1
    return arr[(end - 1 + size) % size];
}

public boolean isFull() {
    return count == size;
}

public int size() {
    return count;
}

public boolean isEmpty() {
    return count == 0;
}

}
文末思考
数组增删头部元素的效率真的只能是 O(N) 么?
我们都说,在数组增删头部元素的时间复杂度是 O(N),因为需要搬移元素。但是,如果我们使用环形数组,其实是可以实现在 O(1) 的时间复杂度内增删头部元素的。
当然,上面实现的这个环形数组只提供了 addFirst, removeFirst, addLast, removeLast 这几个方法,并没有提供 我们之前实现的动态数组 的某些方法,比如删除指定索引的元素,获取指定索引的元素,在指定索引插入元素等等。
但是你可以思考一下,难道环形数组实现不了这些方法么?环形数组实现这些方法,时间复杂度相比普通数组,有退化吗?好像没有吧。环形数组也可以删除指定索引的元素,也要做数据搬移,和普通数组一样,复杂度是 O(N);环形数组也可以获取指定索引的元素(随机访问),只不过不是直接访问对应索引,而是要通过 start 计算出真实索引,但计算和访问的时间复杂度依然是 O(1);环形数组也可以在指定索引插入元素,当然也要做数据搬移,和普通数组一样,复杂度是 O(N)。
你可以思考一下是不是这样。如果是这样,为什么编程语言的标准库中提供的动态数组容器底层并没有用环形数组技巧。

相关文章
|
1天前
|
存储 缓存 算法
学习数据结构和算法的框架思维
本文系统梳理数据结构与算法核心思想:所有数据结构本质为数组或链表的变形,基本操作均为遍历与访问;算法本质是穷举,关键在于“无遗漏”和“无冗余”。掌握框架思维,方能以不变应万变,高效刷题。
学习数据结构和算法的框架思维
|
1天前
|
存储 数据可视化 Java
用拉链法实现哈希表
本文详解哈希表中拉链法的实现原理,通过简化版与完整版Java代码,介绍如何用链表解决哈希冲突,支持泛型、动态扩容及增删查改操作,帮助深入理解哈希表底层机制。
|
1天前
|
存储 缓存 算法
哈希表核心原理
哈希表不等于Map。Map是键值映射的抽象接口,哈希表(如HashMap)是其基于数组和哈希函数的具体实现之一。增删查改O(1)的性能依赖于哈希函数效率与冲突处理,而Map其他实现(如TreeMap)复杂度可能为O(logN)。需注意哈希冲突、扩容、负载因子及key不可变性等核心问题。
|
1天前
|
存储 算法 Java
动态数组代码实现
本文详解动态数组的底层实现,涵盖自动扩缩容、索引越界检查与内存泄漏防范三大关键点,结合Java代码演示增删查改操作及扩容机制,帮助理解数据结构设计原理。
|
1天前
|
算法 数据可视化
二叉树的递归/层序遍历 递归遍历(DFS)
本文详解二叉树的遍历框架,涵盖递归遍历的固定访问顺序及前、中、后序的本质区别——即代码在递归函数中的位置不同所致。同时深入讲解层序遍历(BFS)的三种实现方式,适用于不同场景,尤其适合求最短路径问题;而DFS则因结构天然适合搜索所有路径。通过实例对比,阐明BFS与DFS的应用偏好及原理依据。
二叉树的递归/层序遍历 递归遍历(DFS)
|
1天前
|
人工智能 Java 程序员
JavaSE进阶
本文介绍了Java开发入门的完整流程,涵盖JDK安装、IDEA配置与使用、第一个Java程序的创建与运行。内容包括项目搭建、模块与包的创建、代码注释规范、常用快捷键及通义灵码插件安装等实用技巧,并结合真实工作场景给出操作建议,适合初学者快速掌握开发环境配置与基础编码技能。(239字)
JavaSE进阶
多叉树的递归/层序遍历
多叉树是二叉树的扩展,每个节点可有多个子节点。遍历方式类似:递归遍历无中序概念;层序遍历用队列实现,可记录深度或适配加权边,代码结构与二叉树一致,仅子节点处理由左右变为列表遍历。
|
1天前
|
存储 算法 索引
二叉树基础及常见类型
二叉树是最核心的数据结构之一,不仅是红黑树、堆、字典树等复杂结构的基础,更体现了递归思维的本质。掌握二叉树,等于掌握算法解题的钥匙。从满二叉树到完全二叉树,再到二叉搜索树,各类变体应用广泛。其链式存储与哈希表表示法在算法题中灵活实用,是刷题进阶的必经之路。
|
1天前
|
算法 Python
双端队列(Deque)原理及实现
双端队列支持在队头和队尾高效地插入、删除元素,时间复杂度均为O(1)。相比标准队列的“先进先出”,它更灵活,类似两端可进出的过街天桥。可用链表或环形数组实现,常用于算法题中模拟栈或队列。
|
1天前
|
Java API
用数组实现队列/栈
使用数组实现栈时,可将动态数组尾部作为栈顶,利用ArrayList的add和remove方法实现O(1)时间复杂度的入栈、出栈操作。若以头部为栈顶,则需借助环形数组(如CycleArray)实现高效操作。同样,基于环形数组还可轻松实现队列,通过addLast入队、removeFirst出队,满足队列先进先出特性,所有操作均保持O(1)时间复杂度。