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

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
ListNode *removeNthFromEnd(ListNode *head, int n) {
    struct ListNode *q=0,*w=0,*e=0;
    int i,j=0;
    if(n==0)    return head;    //无需修改
    if(n==1&&head->next==0)        return 0;    //只有一个结点，并且要删除掉它
    q=e=head;                //e为最后一个结点的位置
    for(i=0;i<n;i++){            //e先设置在q的后面为n个距离的地方,q->next是要删的结点
        e=e->next;
        j++;
        if(e->next==0)    //针对刚好要求删掉链表的头结点
            break;
    }
    while(e->next!=0){        //将e和q保持相同距离，往后移直到e指向最后一个元素，q->next就是所要删除的结点了
        e=e->next;    
        q=q->next;
    }
    if(j<n)    // 对于需要删除头结点的情况，用j判断
        return head->next;
    else
        q->next=q->next->next;
    return head;
}
};

 

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

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

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