问题描述:
给定一个链表,删除链表中倒数第n个节点,返回链表的头节点。
样例:
给出链表1->2->3->4->5->null和 n = 2.删除倒数第二个节点之后,这个链表将变成1->2->3->5->null。
思路:
用一个循环数清楚这个链表有几个元素,在循环找到倒数第n+1个元素和倒数第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) {
// write your code here
ListNode *p= new ListNode(0);
ListNode *q= new ListNode(0);
p= head;
int count= 1;
while (p->next!= NULL )
{ p= p->next;
count++;
}
if ((count-n)== 0) {
head= head->next;
return head;
}
p= head;
for (int i= 1; i< count-n+1; i++)
{ q= p;
p= p->next;
}
q->next= p->next;
delete p;
return head;
}
};