单链表

单链表实例

#include <stdio.h>

typedef struct sksstats_task_node{
    unsigned char taskId;
    unsigned int data;
    struct sksstats_task_node* next;
}SKSSTATS_TASK_NODE;

SKSSTATS_TASK_NODE* g_pTaskHead = NULL;

SKSSTATS_TASK_NODE* SksStatsCreateTaskNode(unsigned char taskId)
{
    SKSSTATS_TASK_NODE* p = NULL;

    p = (SKSSTATS_TASK_NODE*)malloc(sizeof(SKSSTATS_TASK_NODE));
    if(NULL == p)
    {
        printf("have no memory for saving task!\n");
        return NULL;
    }
    memset(p, 0, sizeof(SKSSTATS_TASK_NODE));

    p->taskId = taskId;
    p->next = NULL;

    return p;
}

SKSSTATS_TASK_NODE* SksStatsFindTaskNode(SKSSTATS_TASK_NODE* pTaskHead, unsigned char taskId)
{
    if(pTaskHead == NULL)
    {
        return NULL;
    }
    if(pTaskHead->taskId == taskId)
    {
        return pTaskHead;
    }

    return SksStatsFindTaskNode(pTaskHead->next, taskId);
}

int SksStatsDeleteTaskNode(SKSSTATS_TASK_NODE** ppTaskNodeCur, SKSSTATS_TASK_NODE* pTaskNode)
{
    SKSSTATS_TASK_NODE* pTaskNodePre = NULL;
    SKSSTATS_TASK_NODE* pTaskNodeTmp = NULL;

    if(ppTaskNodeCur == NULL || *ppTaskNodeCur == NULL || pTaskNode == NULL)
    {
        return 1;
    }

    //删除头结点
    if(*ppTaskNodeCur == pTaskNode)//也可以通过比较taskId
    {
        *ppTaskNodeCur = (*ppTaskNodeCur)->next;
        free(pTaskNode);
        return 0;
    }

    //删除头结点后的节点
    pTaskNodePre = *ppTaskNodeCur;
    while(pTaskNodePre->next != NULL)
    {
        pTaskNodeTmp = pTaskNodePre->next;
        if(pTaskNodeTmp == pTaskNode)//也可以通过比较taskId
        {
            pTaskNodePre->next = pTaskNodeTmp->next;
            free(pTaskNode);
            return 0;
        }
        pTaskNodePre = pTaskNodePre->next;
    }

    return 2;
}

int SksStatsAddTaskNode(SKSSTATS_TASK_NODE** ppTaskNodeCur, SKSSTATS_TASK_NODE* pTaskNode)
{
    if(NULL == ppTaskNodeCur)
    {
        return 1;
    }

    if(NULL == *ppTaskNodeCur)
    {
        *ppTaskNodeCur = pTaskNode;
        return 0;
    }

    return SksStatsAddTaskNode(&(*ppTaskNodeCur)->next, pTaskNode);
}

int main(int argc, char *argv[])
{
    unsigned char taskId = 0;
    SKSSTATS_TASK_NODE* p = NULL;
    SKSSTATS_TASK_NODE* d = NULL;

    if(argc != 2)
    {
        printf("param error!\n");
        return 1;
    }
    taskId = argv[1];

    /* 删除相同taskId的任务 */
    d = SksStatsFindTaskNode(g_pTaskHead, taskId);
    if(d != NULL)
    {
        if(SksStatsDeleteTaskNode(&g_pTaskHead, d) != 0)
        {
            printf("delete task failed!\n");
        }
    }

    p = SksStatsCreateTaskNode(taskId);
    if(p == NULL)
    {
        return 2;
    }
    /* 加入TaskNode链表 */
    if(SksStatsAddTaskNode(&g_pTaskHead, p) != 0)
    {
        printf("add task failed!\n");
        free(p);
        return 3;
    }

    return 0;
}