首页 > 代码库 > LeetCode Remove Nth Node From End of List 删除链表的倒数第n个结点

LeetCode Remove Nth Node From End of List 删除链表的倒数第n个结点

 1 /** 2  * Definition for singly-linked list. 3  * struct ListNode { 4  *     int val; 5  *     ListNode *next; 6  *     ListNode(int x) : val(x), next(NULL) {} 7  * }; 8  */ 9 class Solution {10 public:11 ListNode *removeNthFromEnd(ListNode *head, int n) {12     struct ListNode *q=0,*w=0,*e=0;13     int i,j=0;14     if(n==0)    return head;    //无需修改15     if(n==1&&head->next==0)        return 0;    //只有一个结点,并且要删除掉它16     q=e=head;                //e为最后一个结点的位置17     for(i=0;i<n;i++){            //e先设置在q的后面为n个距离的地方,q->next是要删的结点18         e=e->next;19         j++;20         if(e->next==0)    //针对刚好要求删掉链表的头结点21             break;22     }23     while(e->next!=0){        //将e和q保持相同距离,往后移直到e指向最后一个元素,q->next就是所要删除的结点了24         e=e->next;    25         q=q->next;26     }27     if(j<n)    // 对于需要删除头结点的情况,用j判断28         return head->next;29     else30         q->next=q->next->next;31     return head;32 }33 };

 

题意:给一个链表,要删除从链尾数起,第n个结点,然后返回此链。

思路:分几种情况,一是要删除尾结点,即n=1;二是要删除头结点,需要特殊判断,同时又要缩小代码量(若加多几行代码就容易写多了)。

测试一直出问题,没考虑好只有两个结点的情况。当有两个结点的链表时,n=1或n=2的情况处理好了,后面一般没问题了。

LeetCode Remove Nth Node From End of List 删除链表的倒数第n个结点