数据结构——单链表操作代码实现

单链表操作代码实现

//单链表结点定义
typedef struct node{
    int data;
    struct node *next;
}NODE,*LinkList;

//单链表查找,在表中查找第k个元素，若找到，返回该节点的指针，否则返回NULL
//L为带头结点单链表的头指针
LinkList Find_List(LinkList L,int k)
{
    LinkList p;
    int i = 1;
    p = L->next;
    while (p && i < k) {
        p = p->next;
        i++;
    }
    
    if(p && i == k){
        return p;
    }
    
    return NULL;
}

//单链表插入，将newElem插入表中的第k个元素之前，成功返回0，否则返回-1
//L为带头结点单链表的头指针
int Insert_List(LinkList L,int k,int newElem)
{
    LinkList p,s;
    if(k == 1){
        p = L;
    }else{
        p = Find_List(L,k-1);
    }
    if(!p){
        return -1;
    }
    
    s = (NODE*)malloc(sizeof(NODE));//创建新元素的空间
    if(!s){
        return -1;
    }
    s->data = newElem;
    s->next = p->next;
    p->next = s;//等同于将newElem插入第k-1个元素之后
    return 0;
}

//单链表的删除,删除表中第k个元素，成功返回0，否则返回-1
//L为带头结点单链表的头指针
//删除第k个元素等同于将第k-1个元素的next指向第k+1个元素
int Delete_List(LinkList L,int k)
{
    LinkList p;
    if(k == 1){
        p = L;
    }else{
        p = Find_List(L,k-1);
    }

    if(!p || !p->next){
        return -1;
    }
    p->next = p->next->next;
    return 0;
}