栈(stack)是一种遵循先入后出逻辑的线性数据结构。
如图所示,我们将堆叠元素的顶部称为“栈顶”,底部称为“栈底”。将把元素添加到栈顶的操作叫做“入栈”,删除栈顶元素的操作叫做“出栈”。

1. 栈的常用操作
栈的常用操作如下表所示:
方法 |
描述 |
时间复杂度 |
push() |
元素入栈(添加至栈顶) |
$O(1)$ |
pop() |
栈顶元素出栈 |
$O(1)$ |
peek() |
访问栈顶元素 |
$O(1)$ |
通常情况下,一些编程语言内置了栈类。当没有提供栈类时,可以将该语言的“数组”或“链表”当作栈来使用,并在程序逻辑上忽略与栈无关的操作。C++具有栈类stack可以直接使用,而Python的列表也有能轻易实现栈操作的方法,因此下面以C为例实现栈。
2. 栈的实现
栈可以视为一种受限制的数组或链表,因为它只能在栈顶添加或删除元素。换句话说,我们可以通过屏蔽数组或链表的部分无关操作,使其对外表现的逻辑符合栈的特性。
2.1 基于链表的实现
使用链表实现栈时,可以将链表的头节点视为栈顶,尾节点视为栈底。对于入栈操作,只需将元素插入链表头部,这种节点插入方法被称为“头插法”。而对于出栈操作,只需将头节点从链表中删除即可。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62
| typedef struct { ListNode *top; int size; } LinkedListStack;
LinkedListStack *newLinkedListStack() { LinkedListStack *s = malloc(sizeof(LinkedListStack)); s->top = NULL; s->size = 0; return s; }
void delLinkedListStack(LinkedListStack *s) { while (s->top) { ListNode *n = s->top->next; free(s->top); s->top = n; } free(s); }
int size(LinkedListStack *s) { return s->size; }
bool isEmpty(LinkedListStack *s) { return size(s) == 0; }
void push(LinkedListStack *s, int num) { ListNode *node = (ListNode *)malloc(sizeof(ListNode)); node->next = s->top; node->val = num; s->top = node; s->size++; }
int peek(LinkedListStack *s) { if (s->size == 0) { printf("栈为空\n"); return INT_MAX; } return s->top->val; }
int pop(LinkedListStack *s) { int val = peek(s); ListNode *tmp = s->top; s->top = s->top->next; free(tmp); s->size--; return val; }
|
2.2 基于数组的实现
在使用数组实现栈时,我们可以将数组的尾部视为栈顶。入栈与出栈操作分别对应在数组尾部添加元素与删除变量,时间复杂度都为$O(1)$。
由于入栈的元素可能会源源不断地增加,因此我们可以使用动态数组,这样就无需自行处理数组扩容问题。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56
| typedef struct { int *data; int size; } ArrayStack;
ArrayStack *newArrayStack() { ArrayStack *stack = malloc(sizeof(ArrayStack)); stack->data = malloc(sizeof(int) * MAX_SIZE); stack->size = 0; return stack; }
void delArrayStack(ArrayStack *stack) { free(stack->data); free(stack); }
int size(ArrayStack *stack) { return stack->size; }
bool isEmpty(ArrayStack *stack) { return stack->size == 0; }
void push(ArrayStack *stack, int num) { if (stack->size == MAX_SIZE) { printf("栈已满\n"); return; } stack->data[stack->size] = num; stack->size++; }
int peek(ArrayStack *stack) { if (stack->size == 0) { printf("栈为空\n"); return INT_MAX; } return stack->data[stack->size - 1]; }
int pop(ArrayStack *stack) { int val = peek(stack); stack->size--; return val; }
|
3. 两种实现的对比
3.1 时间效率
在基于数组的实现中,入栈和出栈操作都在预先分配好的连续内存中进行,具有很好的缓存本地性,因此效率较高。然而,如果入栈时超出数组容量,会触发扩容机制,导致该次入栈操作的时间复杂度变为$O(n)$。
在基于链表的实现中,链表的扩容非常灵活,不存在上述数组扩容时效率降低的问题。但是,入栈操作需要初始化节点对象并修改指针,因此效率相对较低。不过,如果入栈元素本身就是节点对象,那么可以省去初始化步骤,从而提高效率。
综上所述,当入栈与出栈操作的元素是基本数据类型时,例如int
或double
,有如下结论:
- 基于数组实现的栈在触发扩容时效率会降低,但由于扩容是低频操作,因此平均效率更高。
- 基于链表实现的栈可以提供更加稳定的效率表现。
3.2 空间效率
在初始化列表时,系统会为列表分配“初始容量”,该容量可能超出实际需求;并且,扩容机制通常是按照特定倍率(例如2倍)进行扩容的,扩容后的容量也可能超出实际需求。因此,基于数组实现的栈可能造成一定的空间浪费。
然而,由于链表节点需要额外存储指针,因此链表节点占用的空间相对较大。故两者的空间效率仍需看情况讨论。
参考
[1] GitHub 开源项目《hello 算法》
[2] 程杰.大话数据结构【溢彩加强版】[M].清华大学出版社,2020.