1.两次遍历:先遍历一次得出链表长度,再遍历一次删除节点,需要注意删除头节点的情况。
/**
* 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) {
ListNode* tmp = head, *result = head;
int l = 0;
while (tmp != NULL) {
l++;
tmp = tmp->next;
}
tmp = head;
if (n == l) {
result = head->next;
} else {
int num = 1;
int tar_index = l - n;
while (num != tar_index) {
tmp = tmp->next;
num++;
}
tmp->next = tmp->next->next;
}
return result;
}
};
2.一次遍历,在链表头部加入了一个节点,使得删除头节点的操作不用特殊处理。主要是用了双指针的方法,先将第一个指针往前推n+1个,之后再将第一个和第二个指针同时向后推,直到第一个指针为控,这样能保证这两个指针之间永远相隔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) {
ListNode* tmpNode = new ListNode(0);
tmpNode->next = head;
ListNode* tmp1 = tmpNode, *tmp2 = tmpNode;
int num = 0;
while (num < n + 1) {
tmp1 = tmp1->next;
num++;
}
while (tmp1 != NULL) {
tmp1 = tmp1->next;
tmp2 = tmp2->next;
}
tmp2->next = tmp2->next->next;
return tmpNode->next;
}
};