给定一个链表,删除链表中倒数第n个节点,返回链表的头节点。
样例
给出链表1->2->3->4->5->null和 n = 2.
删除倒数第二个节点之后,这个链表将变成1->2->3->5->null.
挑战
O(n)时间复杂度
要考虑的测试用例情况(len表示链表长度):
1. n>len,
2. 要删除的结点是头节点或只有一个结点(只有一个结点要特殊处理,因为没有前驱也没有后继)
3. 要删除的结点是最后一个结点(要从头开始找到最后一个结点前的一个结点)
4. 要删除的结点在链表中间(不是最后一个也不是第一个)
我的想法:
首先遍历一遍链表,计算链表的长度(for语句停止条件注意),然后找到倒数第n个待删除结点,倒数第n个结点和第n+1个结点值交换一下,删除倒数第n+1个结点。如果n=1,则需重新遍历一遍,找到前驱,特殊处理。复杂度O(n),但是基本上要遍历2次。
/**
* 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
if(head == NULL){
return NULL;
}
if(n<=0)
{
return nullptr;
}
int len = 0;
ListNode *p = head;
for(p; p!=NULL; p= p->next){
len++;
}
if(len == 1&& n == 1){
return NULL;
}
p = head;
for(int i = 1; i<=len-n; i++){
p = p->next;
}//p at nth node
if(p->next == NULL){
ListNode *pre;
for(pre = head;pre->next!=p; pre = pre->next);
pre->next = NULL;
return head;
}
ListNode *q = p->next;
int tmp = q->val;
q->val = p->val;
p->val = tmp;
p->next = q->next;
return head;
}
};
然而这题实际上有更好的解法:双指针,一个指针先走n-1步
/**
* 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
if(head == NULL||n<=0){
return NULL;
}
ListNode *pHead = head;
for(int i = 0; i< n-1; i++){ //中途可能会断掉,len<n的情况
pHead = pHead ->next;
if(pHead == NULL){//提前返回
return NULL;
}
}
ListNode *q = head;
if(q->next == NULL){//只有一个结点,直接返回
return NULL;
}
while(pHead->next!=NULL){
pHead = pHead->next;
q = q->next;
}
if(q->next == NULL){//最后一个结点,再遍历
ListNode *pre = head;
while(pre->next != q){
pre = pre->next;
}
pre->next = NULL;
return head;
}else{
int tmp = q->next->val;
q->next->val = q->val;
q->val = tmp;
q->next = q->next->next;
return head;
}
}
};