码迷,mamicode.com
首页 > 其他好文 > 详细

Remove Nth Node From End of List

时间:2014-11-14 22:27:28      阅读:175      评论:0      收藏:0      [点我收藏+]

标签:style   blog   http   io   color   ar   os   使用   sp   

Given a linked list, remove the nth 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.

 

由于删除头结点时要改变头结点的指针,所以传入的是引用或者使用二级指针。提示中给的只是传入一级指针,要是删除头结点,始终不能删掉。(注意要改变一个数据结构时,要传二级指针或者引用,树中的很多操作都是这样的)

C++实现代码如下:

#include<iostream>
#include<new>
using namespace std;

//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)
    {
        if(head==NULL)
            return NULL;
        ListNode *p=head;
        ListNode *q=head;
        ListNode *pre=head;
        int i=0;
        while(q&&i<n)
        {
            q=q->next;
            i++;
        }
        //说明没有n个元素
        if(i<n)
           return head;
        while(q)
        {
            pre=p;
            p=p->next;
            q=q->next;
        }
        if(p==head)
        {
            head=head->next;
            delete p;
        }
        else
        {
            pre->next=p->next;
            delete p;
        }
        return head;
    }
    void createList(ListNode *&head)
    {
        ListNode *p=NULL;
        int i=0;
        int arr[10]= {10,9,8,7,6,5,4,3,2,1};
        for(i=0; i<5; i++)
        {
            if(head==NULL)
            {
                head=new ListNode(arr[i]);
                if(head==NULL)
                    return;
            }
            else
            {
                p=new ListNode(arr[i]);
                p->next=head;
                head=p;
            }
        }
    }
};

int main()
{
    Solution s;
    ListNode *L=NULL;
    s.createList(L);
    ListNode *head=L;
    while(head)
    {
        cout<<head->val<<" ";
        head=head->next;
    }
    cout<<endl;
    s.removeNthFromEnd(L,5);
    while(L)
    {
        cout<<L->val<<" ";
        L=L->next;
    }
}

运行结果:

bubuko.com,布布扣

 

Remove Nth Node From End of List

标签:style   blog   http   io   color   ar   os   使用   sp   

原文地址:http://www.cnblogs.com/wuchanming/p/4098255.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!