(关注数据结构和算法,了解更多新知识)
在网上看到一网友说一个女程序员被逼哭了,原因是有很多事不知道如何下手,导致压力过大,绷不住了。其实不光是女程序员,任何行业任何一个人都一样,工作最怕的就是事很多但又不知道从何做起,导致非常着急,压力大。
--------------下面是今天的算法题--------------
来看下今天的算法题,这题是LeetCode的第19题:删除链表的倒数第 N 个结点。
问题描述
来源:LeetCode第19题
难度:中等
给你一个链表,删除链表的倒数第 n 个结点,并且返回链表的头结点。
示例1:
输入:head = [1,2,3,4,5], n = 2
输出:[1,2,3,5]
示例2:
输入:head = [1], n = 1
输出:[]
链表中结点的数目为 sz
1 <= sz <= 30
0 <= Node.val <= 100
1 <= n <= sz
问题分析
这题是让删除链表的倒数第 n 个节点,如果要删除一个节点,必须要知道他的前一个节点(头节点除外),所以如果要删除倒数第 n 个节点,我们必须要找到倒数第 n+1 个节点。
我们知道在单向链表中是没法从后往前遍历的,怎么找到倒数第 n+1 个节点呢?这里有两种方式,一种是使用两个指针,刚开始的时候这两个指针都指向头节点,其中一个指针先走 n 步,然后两个指针在同时走,当先走的指针指向空的时候,后走的指针指向的就是倒数第 n+1 个节点,这种方式代码比较简单,就不在写了。
我们来看另一种方式,就是使用递归找到倒数第 n+1 个节点。因为递归实际上是分为两部分,分别是递和归,递就是传递的意思,往下走,归就是回去的意思,往回走。我们只需要在往回走的时候统计节点的个数,就相当于链表从后往前数了,当数到第 n+1 个的时候就把第 n 个节点给删除。
这里要注意要找的是倒数第 n+1 个节点,如果删除的是头节点,那么头节点前面是没有节点的,所以这个时候是找不到倒数第 n+1 个节点的,所以如果删除的是头节点,需要单独处理。
JAVA:
public ListNode removeNthFromEnd(ListNode head, int n) {
int len = length(head, n);// 获取链表的长度,随便把节点删除。
if (len == n) // 说明删除的是头节点。
return head.next;
return head;
}
// 获取节点的长度,从后往前数,到第n+1个节点的时候把倒数第n个节点删除。
private int length(ListNode node, int n) {
if (node == null)
return 0;
int len = length(node.next, n) + 1;// 计算链表长度。
// 获取倒数第n+1个节点,就可以完成链表的删除。
if (len == n + 1)
node.next = node.next.next;
return len;
}
C++:
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
int len = length(head, n);// 获取链表的长度,随便把节点删除。
if (len == n) // 说明删除的是头节点。
return head->next;
return head;
}
// 获取节点的长度,从后往前数,到第n+1个节点的时候把倒数第n个节点删除。
int length(ListNode* node, int n) {
if (!node)
return 0;
int len = length(node->next, n) + 1;// 计算链表长度。
// 获取倒数第n+1个节点,就可以完成链表的删除。
if (len == n + 1)
node->next = node->next->next;
return len;
}
C:
// 获取节点的长度,从后往前数,到第n+1个节点的时候把倒数第n个节点删除。
int length(struct ListNode* node, int n) {
if (!node)
return 0;
int len = length(node->next, n) + 1;// 计算链表长度。
// 获取倒数第n+1个节点,就可以完成链表的删除。
if (len == n + 1)
node->next = node->next->next;
return len;
}
struct ListNode* removeNthFromEnd(struct ListNode* head, int n) {
int len = length(head, n);// 获取链表的长度,随便把节点删除。
if (len == n) // 说明删除的是头节点。
return head->next;
return head;
}
Python:
def removeNthFromEnd(self, head: Optional[ListNode], n: int) -> Optional[ListNode]:
# 获取节点的长度,从后往前数,到第n + 1个节点的时候把倒数第n个节点删除。
def length(node: ListNode, n: int) -> int:
if not node:
return 0
len = length(node.next, n) + 1 # 计算链表长度。
# 获取倒数第n + 1个节点,就可以完成链表的删除。
if len == n + 1:
node.next = node.next.next
return len
len = length(head, n) # 获取链表的长度,随便把节点删除。
if len == n: # 说明删除的是头节点。
return head.next
return head
笔者简介
博哥,真名:王一博,毕业十多年,《算法秘籍》作者,专注于数据结构和算法的讲解,在全球30多个算法网站中累计做题2000多道,在公众号中写算法题解800多题,对算法题有自己独特的解题思路和解题技巧,喜欢的可以给个关注,也可以下载我整理的1000多页的PDF算法文档。