在单链表和双链表中删除倒数第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 ||  > lastKth)
         return head;
     SingleNode *cur = head;
     while( NULL != cur)
     {
         lastKth--;
         cur = cur->next;
     }
     if( == lastKth)
     {
         cur = head;
         head = head->next;
         delete cur;
     }
     if( > lastKth)
     {
         cur = head;
         while(++lastKth != )
             cur = cur->next;
         SingleNode *tmp = cur->next;
         cur->next = tmp->next;
         delete tmp;
     }
     return head;
 }

 DoubleNode* removeLastKthNode(DoubleNode *head,int lastKth)
 {
     if(NULL == head ||  > lastKth)
         return head;
     DoubleNode *cur = head;
     while( NULL != cur)
     {
         lastKth--;
         cur = cur->next;
     }
     if( == lastKth)
     {
         cur = head;
         head = head->next;
         head->pre = NULL;
         delete cur;
     }
     if( > lastKth)
     {
         cur = head;
         while(++lastKth != )
             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 =;i<;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 = ;
     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 ;
 }

【说明】：

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

注：

　　转载请注明出处；

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