删除链表中倒数第n个节点

一.题目描述

给定一个链表，删除链表中倒数第n个节点，返回链表的头节点。样例:给出链表1->2->3->4->5->null和 n = 2.

删除倒数第二个节点之后，这个链表将变成1->2->3->5->null.

二.解题思路

定义两个指针,让一个指针先走,走到第n个节点时另一个指针再开始走,当第一个指针走到最后一个节点时,因为第二个与之相差n步,所以第二个指针在倒数第n+1个节点,然后将倒数第n个节点覆盖,返回头结点.

三.代码实现

/**
 * Definition of ListNode
 * class ListNode {
 * public:
 *     int val;
 *     ListNode *next;
 *     ListNode(int val) {
 *         this->val = val;
 *         this->next = NULL;
 *     }
 * }
 */
class Solution {
public:
    /**
     * @param head: The first node of linked list.
     * @param n: An integer.
     * @return: The head of linked list.
     */
    ListNode *removeNthFromEnd(ListNode *head, int n) {
        // write your code here

       ListNode *p=new ListNode;
       ListNode *f=new ListNode;
       p->next=head;
       if(n<1) return head;
       
       while(--n>=0)
       {  p = p->next;   } 
       
       if(p->next==NULL) 
       { return head->next;}
       
       f->next=head;  
       while(p->next!=NULL)
       {  
          p = p->next;  
          f = f->next;
        }  
       f->next=f->next->next;
       return head;  
    }
};

四.感想

考虑问题要全面,n小于1时说明不需要删元素,直接返回头结点即可;如果需要删除的元素是头结点,则需要返回链表的第二个节点.因为一开始没有考虑到删除的结点是第一个结点的情况,所以wrong answer了好多遍.