leetcode Remove Nth Node From End of List

Given a linked list, remove the n^th node from the end of list and return its head.

For example,

   Given linked list: 1->2->3->4->5, and n = 2.

   After removing the second node from the end, the linked list becomes 1->2->3->5.

Note:
Given n will always be valid.
Try to do this in one pass. 

 

 

PS：大早上遇到一道简单题还是非常令人开心的。

非常简单，首先遍历得到length。然后遍历至length-n（也就是要删除的节点）删除就行了。

注意两个地方，假如删除的是头节点，也就是length==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) {
        int length=0;
        ListNode* p=head,*q;
        while(p!=NULL){
            p=p->next;
            length++;
        }
        if(length==n){
            if(head->next!=NULL) {
                head=head->next;
                return head;}
            else return NULL;
        }
        p=head;
        q=p;
        for(int i=0;i<length-n;i++){
            q=p;
            p=p->next; 
        }
        if(p->next!=NULL) q->next=p->next;
        else q->next=NULL;
        return head;
    }
};