您的位置:首页 > 其它

删除链表中倒数第n个元素

2017-03-29 23:21 239 查看
问题描述:

给定一个链表,删除链表中倒数第n个节点,返回链表的头节点。

样例

给出链表1->2->3->4->5->null和
n = 2.

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

解题思路:

建立两个指针,使两个指针间隔为n,前一个指针走到NULL时,后一个指针走到倒数第n个节点的位置,这时让倒数第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){
        ListNode*h1=head;
        ListNode*h2=head;
        int count=0;
        while(head!=NULL){
            head=head->next;
            count++;
        }
        if(count<n) return h1;
        if(count>n){
            for(int i=1;i<(count-n);i++){
                h2=h2->next;
            }
            h2->next=h2->next->next;
            return h1;
        }
        if(count==n){
            return h1->next;
        }
    }
};
解题感悟:

忘了判断链表长度小于n的情况,如果链表长度小于n,则直接返回h1。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  链表