问题描述:
给定一个链表,删除链表中倒数第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。