在单链表和双链表中删除倒数第K个节点

【说明】：

　　本文是左程云老师所著的《程序员面试代码指南》第二章中“在单链表和双链表中删除倒数第K个节点”这一题目的C++复现。

　　本文只包含问题描述、C++代码的实现以及简单的思路，不包含解析说明，具体的问题解析请参考原书。

　　感谢左程云老师的支持。

【题目】：

　　分别实现两个函数，一个可以删除单链表中倒数第 K 个节点，另一个可以删除双链表中倒数第 K 个节点。

【要求】：

　　如果链表长度为 N，时间复杂度达到 O(N)，额外空间复杂度达到 O(1)。

 【思路】：

　　在确定待删除节点的位置有一个小技巧，大家可以看代码推断，也可以翻看左老师的原书奥。

【编译环境】：

　　CentOS6.7（x86_64）

　　gcc 4.4.7

 【实现】：

　　实现及测试代码：

/*
 *文件名：lists_delLastKth.cpp
 *作者：
 *摘要：删除单链表或双链表的倒数第 K 个节点
 */

#include <iostream>

using namespace std;

struct SingleNode
{
    int value;
    SingleNode *next;
};

struct DoubleNode
{
    int value;
    DoubleNode *pre;
    DoubleNode *next;
};

SingleNode* removeLastKthNode(SingleNode *head,int lastKth)
{
    if(NULL == head || 1 > lastKth)
        return head;
    SingleNode *cur = head;
    while( NULL != cur)
    {
        lastKth--;
        cur = cur->next;
    }
    if(0 == lastKth)
    {
        cur = head;
        head = head->next;
        delete cur;
    }
    if(0 > lastKth)
    {
        cur = head;
        while(++lastKth != 0)
            cur = cur->next;
        SingleNode *tmp = cur->next;
        cur->next = tmp->next;
        delete tmp;
    }
    return head;        
}

DoubleNode* removeLastKthNode(DoubleNode *head,int lastKth)
{
    if(NULL == head || 1 > lastKth)
        return head;
    DoubleNode *cur = head;
    while( NULL != cur)
    {
        lastKth--;
        cur = cur->next;
    }
    if(0 == lastKth)
    {
        cur = head;
        head = head->next;
        head->pre = NULL;
        delete cur;
    }
    if(0 > lastKth)
    {
        cur = head;
        while(++lastKth != 0)
            cur = cur->next;
        DoubleNode *tmp = cur->next;
        cur->next = tmp->next;
        if(NULL != tmp->next)
            tmp->next->pre = cur;
        delete tmp;
    }
    return head;        
}

int main()
{
    SingleNode *shead = NULL;
    SingleNode *sptr;
    DoubleNode *dhead = NULL;
    DoubleNode *dptr;
    for(int i =0;i<10;i++)
    {
        if(NULL == shead && NULL == dhead)
        {    
            //单链表
            shead = new SingleNode;
            shead->value = i;
            shead->next = NULL;
            sptr = shead;
            //双链表
            dhead = new DoubleNode;
            dhead->value = i;
            dhead->next = NULL;
            dhead->pre = NULL;
            dptr = dhead;
            continue;
        }
        //单链表
        sptr->next = new SingleNode;
        sptr = sptr->next;
        sptr->value = i;
        sptr->next = NULL;
        //双链表
        dptr->next = new DoubleNode;
        dptr->next->pre = dptr;
        dptr = dptr->next;
        dptr->value = i;
        dptr->next = NULL;
    }
    int k = 5;
    cout << "Single linked list before remove last " << k << "th data: " << endl;
    sptr = shead;
    while(NULL != sptr)
    {
        cout << sptr->value << " ";
        sptr = sptr->next;
    }
    cout << endl;
    cout << "Double linked list before remove last " << k << "th data: " << endl;
    dptr = dhead;
    while(NULL != dptr)
    {
        cout << dptr->value << " ";
        dptr = dptr->next;
    }
    cout << endl;
    
    removeLastKthNode(shead,k);
    removeLastKthNode(dhead,k);

    sptr = shead;
    dptr = dhead;
    cout << "Single linked list after removed: " << endl;    
    while(NULL != sptr)
    {
        cout << sptr->value << " ";
        sptr = sptr->next;
    }
    cout << endl;
    cout << "Double linked list after removed: " << endl;    
    while(NULL != dptr)
    {
        cout << dptr->value << " ";
        dptr = dptr->next;
    }
    cout << endl;
    return 0;
}

【说明】：

　　这个算法不难，我写的测试代码（main 函数）较为麻烦，大家理解哈。

注：

　　转载请注明出处；

　　转载请注明源思路来自于左程云老师的《程序员代码面试指南》。