给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
方法一:两次遍历
/**
* 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) {
int cnt=1;
ListNode* p=head;
ListNode* q;
while(p->next!=NULL){
p=p->next;
cnt++;
}
//添加头节点
ListNode* dummy = new ListNode(0);
dummy->next = head;
int time=cnt-n;
p=dummy;
while(time--){
p=p->next;
}
q=p->next;
p->next=q->next;
delete q;
return dummy->next;
}
};
方法二:一次遍历
在这个方法中需要两个指针。首先,同样的方法添加头节点,接着找出头节点后第n个节点,一直往后挪,直到最后一个,此时前面的指针指向的下一个就是要删除的节点。
/**
* 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) {
int cnt=1;
int i,j;
//添加头节点
ListNode* dummy = new ListNode(0);
dummy->next = head;
ListNode* p=dummy;
ListNode* q=p;
for (i=0;i<n;i++){
q=q->next;
}
while(q->next!=NULL){
p=p->next;
q=q->next;
}
ListNode* tmp=p->next;
p->next=tmp->next;
delete tmp;
return dummy->next;
}
};