LeetCode-Remove Nth Node From End of List

原创于 2014-09-12 00:07:00 发布 · 381 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#List #leetcode

LeetCode 专栏收录该内容

21 篇文章

订阅专栏

ListNode *removeNthFromEnd(ListNode *head, int n)
{
    if(head==NULL || (head->next==NULL && n==1))
        return NULL;

    
    ListNode *pNode = head;
    int length=0;
    
    while(pNode!=NULL)
    {
        ++length;
        pNode=pNode->next;
    }
    
    if(n>length)
        return head;
    if(n==length && head->next!=NULL)
        return head->next;
    
    int rest=length-n-1;
    pNode=head;
    
    while(rest-- && pNode!=NULL)
    {
        pNode=pNode->next;
    }
    
    if(n==1)
        pNode->next=NULL;
    else
        pNode->next=pNode->next->next;
    
    return head;
}