这地方的嵌套关系好复杂orz 大概是LeetCode第237题是剑指第18-1题的子集,但是牛客上没有剑指18-1题却有18-2题,而剑指的18-2题就是之前做过的LeetCode的第83题(Remove Duplicates from Sorted List)的升级版,即第82题(Remove Duplicates from Sorted List II)。
剑指的18-1题是给定链表中的一个节点并删除它,没有限定需要删除的节点是在链表的什么位置,而LeetCode上已经限定了需要删除的节点不是尾节点、整个链表至少有两个节点,于是需要考虑的情况就变少了。下面还是以剑指上的完整版分情况讨论。
对于删除节点的方法,首先按照最正常的思路,就是从头开始遍历链表,找到待删除节点的前一个节点,然后让这个节点的next指向待删节点的下一节点,这样做的时间复杂度是O(n)。另外一种最开始我自己没有想到的方法就是,因为链表节点带有next指针,我们可以直接把待删节点的下一节点复制到待删节点中,然后把复制后的节点的next指向待删节点next的next,这样就可以O(1)时间达到删除节点的效果。如图所示是两种方法的示意图:
接下来讨论待删节点位置的不同情况:
1. 如果待删节点不是最后一个节点,那么可以采用O(1)的办法进行删除
2. 如果待删节点是最后一个节点,它就没有next了,那就只能从头开始遍历整个链表进行删除
3. 如果待删节点不仅是最后一个节点,还是头节点,也就是说这个链表里只有一个元素,那么直接删除头节点即可
由于leetcode上给出的函数是不带原始链表的,所以这里好像不好实现2和3,只放1的代码上来吧:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
void deleteNode(ListNode* node) {
ListNode* next_node = node->next;
node->val = next_node->val;
node->next = next_node->next;
}
};
后来又自己写了个未经测试的完整版,如有不正还请各位大佬指出~
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
void deleteNode(ListNode* head, ListNode* node) {
// 不是尾节点
if (node->next) {
ListNode* next_node = node->next;
node->val = next_node->val;
node->next = next_node->next;
}
// 链表中只有一个节点
else if (node == head) {
head = NULL;
}
// 是尾节点
else {
ListNode* cur_node = head;
while (cur_node->next != node) {
cur_node = cur_node->next;
}
cur_node->next = NULL;
}
}
};