Linux内核链表复用实现栈

我们当然可以根据栈的特性，向实现链表一样实现栈。但是，如果能够复用已经经过实践证明的可靠数据结构来实现栈，不是可以更加高效吗？

so，今天我们就复用Linux内核链表，实现栈这样的数据结构。

要实现的功能很简单，如下（项目中如需更多功能，可自行添加）：

/* stack.h */

#ifndef _STACK_H_
#define _STACK_H_

#include "list.h"

#define get_stack_top(pos, head, member)        \
        list_entry((head)->prev, typeof(*pos), member)

void stack_creat(struct list_head *list);
void stack_push(struct list_head *new, struct list_head *head);
void stack_pop(struct list_head *entry);
int get_satck_size(struct list_head *head);

#endif /* _STACK_H_ */

/*  stack.c  */

#include "stack.h"

void list_del_tail(struct list_head *head)
{
    __list_del(head->prev->prev,head);
}

void stack_creat(struct list_head *list)
{
    INIT_LIST_HEAD(list);
}

void stack_push(struct list_head *new, struct list_head *head)
{
    list_add_tail(new,head);
}

void stack_pop(struct list_head *head)
{
    struct list_head *list = head->prev; /* 保存链表的最后节点 */

    list_del_tail(head); /* 尾删法 */

    INIT_LIST_HEAD(list); /* 重新初始化删除的最后节点，使其指向自身 */

}

int get_satck_size(struct list_head *head)
{
    struct list_head *pos;
    int size = ;

    if (head == NULL) {
        return -;
    }

    list_for_each(pos,head) {
        size++;
    }

    return size;
}

我们先来说，栈的创建：

非常简单，和内核链表一样，仅仅是将链表指针指向自身。

栈的push（入栈）操作：

也非常得简单，根据栈的先进后出特性，我们采用尾插法，这样最后插入的节点也就位于最后，也就是栈顶。

栈的pop（出栈）操作：

出栈只能操作栈顶元素，所以我们使用尾删法，将内核链表的尾部节点删除，就实现了出栈操作，但是内核链表没有直接实现尾删法，不过，我们已经在前面的随笔中对内核链表进行了分析，显然可以利用内核已经实现了的__list_del函数，稍微改变一下参数，就可以实现尾删法了。

获取栈的大小：

原理也非常简单，循环遍历链表，计数增加即可。

得到栈顶元素：

为什么这里我没有使用函数，而是使用宏呢？这和内核链表的逻辑是一致的。因为如果要写成函数，我必须知道使用栈的人定义的数据类型，如果我定义成void *，又不能使用内核链表的list_entry获取容器结构地址的宏了，所以，我将获取栈顶元素设计为宏，这样我可以不定义数据类型，靠用户输入。

现在，我们通过非常简单的一点代码复用内核链表实现了栈，下面看看测试用例：

#include <stdio.h>
#include <stdlib.h>

#include "stack.h"
#include "list.h"

struct person
{
    int age;
    struct list_head list;
};

int main(int argc,char **argv)
{
    int i;
    int num =;
    struct person *p;
    struct person head;
    struct person *pos,*n;

    stack_creat(&head.list);

    p = (struct person *)malloc(sizeof(struct person )*num);

    for (i = ;i < num;i++) {
        p->age = i*;
        stack_push(&p->list,&head.list);
        p++;
    }

    struct person test;
    test.age = ;

    stack_pop(&head.list);
    stack_pop(&head.list);
    stack_push(&test.list,&head.list);

    printf("==========>\n");
    list_for_each_entry_safe_reverse(pos,n,&head.list,list) {
        printf("%p  age = %d\n",pos,pos->age);
    }

    printf("栈顶节点：%p  age = %d\n",get_stack_top(pos,&head.list,list),
                     get_stack_top(pos,&head.list,list)->age);
    printf("size = %d\n",get_satck_size(&head.list));

    return ;
}

运行结果：

通过复用内核链表，可以非常快速高效地实现很多其他数据结构，所以内核链表一定要充分掌握。

增加判断栈是否为空的函数：

bool is_empt_stack(struct list_head *head)
{
    return list_empty(head);
}

c语言中bool可以包含#include <stdbool.h>，c99可以直接使用_Bool。