1.移除单链表中指定位置的节点
题目描述
给定一个链表,删除链表的倒数第 nn 个节点,并且返回链表的头结点。
原题链接
示例1
输入:head = [1,2,3,4,5], n = 2
输出:[1,2,3,5]
输入:head[1], n=1
输出:[]
输入:head = [1,2], n = 1
输出:[1]
进阶要求
尝试用一趟扫描实现
算法1
两次遍历
- 第一次遍历求出链表长度
- 第二次遍历删掉指定的结点
- 注意删掉头结点的特殊情况(这个自己没有考虑到,当提交调错的时候才解决掉这个问题)
时间复杂度
遍历两次链表,空间复杂度为O(L)
空间复杂度
仅需要定义常数个指针变量,故时间复杂度为O(1)
代码
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
int sum = 0;
ListNode* a = head;
// 求一下链表的长度
while (a) {
++sum;
a = a->next;
}
// 判断删掉第一个节点的特殊情况
if (sum == n) {
head = head->next;
return head;
}
ListNode* b = head;
// 遍历删掉指定的节点
for (int i = 0; i < sum - n - 1; i++) {
head = head->next;
}
head->next = head->next->next;
return b;
}
};
算法2
一次遍历
1.在头部之前添加保护点
2.设置两个指针first和second,均指向保护节点
3.first指针先向后移动n + 1个结点
4.然后first和second指针同时向后移动,直到first指针指向空,此时second结点指向的下一个结点需要删除
原理
始终保持两个指针之间间隔n个结点,在first到达终点时,second的下一个结点就是从结尾数第n个结点。
时间复杂度
遍历两次链表,空间复杂度为O(L)
空间复杂度
仅需要定义常数个指针变量,故时间复杂度为O(1)
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode* e_head = new ListNode(0);
e_head->next = head;
ListNode* first = e_head;
ListNode* second = e_head;
n++;
while (n--) {
first = first->next;
}
while (first) {
first = first->next;
second = second->next;
}
second->next = second->next->next;
return e_head->next;
}
};
本文来自博客园,作者:jsqup,转载请注明原文链接:https://www.cnblogs.com/jsqup/p/15689695.html