题目
给定一个链表,删除链表的倒数第 n n n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
函数原型
C
的函数原型:
struct ListNode* removeNthFromEnd(struct ListNode* head, int n){
}
边界判断
struct ListNode* removeNthFromEnd(struct ListNode* head, int n){
}
算法设计:从前扫一遍,从后扫一遍
思路:先计算链表长度(从前扫一遍),再删除倒数第 n 个结点(从后扫一遍)。
typedef struct ListNode Node;
struct ListNode* removeNthFromEnd(struct ListNode* head, int n){
// 设置一个虚拟头节点(哨兵)
Node* H = malloc(sizeof(Node));
H->next = head;
// 两个指针跟随用于删除
Node*p = H->next, *pre = H;
int cnt = 1;
int i = 1;
while(p){
p = p->next;
cnt++; // 计算length
}
p = H->next; // 把p拉回去
while(i<(cnt-n) ){
pre = p;
p = p->next;
i++;
}
// 循环结束,pre位于要删除节点的前驱,p在删除位置
pre->next = p->next;
free(p);
// 释放哨兵
Node *ans = H->next;
free(H), H = NULL;
return ans;
}
- 时间复杂度: Θ ( n ) \Theta(n) Θ(n)
- 空间复杂度:
Θ
(
1
)
\Theta(1)
Θ(1)
算法 = 快慢指针 & 题目特征 = 链表节点间的相对位置或距离
思路:在这个算法中,我们使用两个指针 p 和 q 来定位倒数第 N 个节点。让指针 p 先向后移动 n 步的原因是为了让指针 p 和指针 q 之间的距离保持为 N 步。这样当指针 p 移动到链表的末尾时,指针 q 就正好指向了倒数第 N 个节点的前一个节点。
配套编码:
struct ListNode* removeNthFromEnd(struct ListNode* head, int n){
struct ListNode *p = head, *q = head;
while(p){ // 同时移动指针 p 和 q,直到指针 p 移动到链表的末尾
if(n < 0) q = q -> next;
n--;
p = p -> next; // 让指针 p 先向后移动 n 步
}
// 指针 q 就指向了倒数第 N 个节点的前一个节点
if(n == 0) return head -> next; // 如果 n 的值等于 0,说明要删除的是链表的头节点,直接返回头节点的下一个节点即可
q -> next = q -> next -> next; // 否则,将指针 q 的下一个节点指向 q 的下下个节点,完成节点的删除操作。
return head; // 最后,返回头节点。
}
- 时间复杂度: Θ ( n ) \Theta(n) Θ(n)
- 空间复杂度: Θ ( 1 ) \Theta(1) Θ(1)
删除链表的倒数第 N 个节点、链表中的环检测、链表的中间节点都可以用快慢指针。
这三个问题之所以可以使用快慢指针算法,是因为它们都涉及到了链表中节点之间的相对位置或者距离的问题。
算法设计:递归
思路:m从后开始统计元素个数,如果等于n,即到达倒数第n个,将当前下一节点指向返回个当前上一节点连接,否则返回当前节点连接 递归回溯,来回一次。
int m;
struct ListNode* removeNthFromEnd(struct ListNode* head, int n){
if( head == NULL ){
m = 0;
return head;
}
head->next = removeNthFromEnd(head->next, n);
return ++m==n ? head->next : head;
}
- 时间复杂度: Θ ( n ) \Theta(n) Θ(n)
- 空间复杂度: Θ ( n ) \Theta(n) Θ(n)