方法一:哈希表
对给定的链表进行一次遍历,并用一个哈希集合(HashSet)来存储所有出现过的节点。对给定的链表元素直接进行「相等」比较,实际上是对两个链表元素的地址(而不是值)进行比较。因此,我们在哈希集合中存储链表元素的值,方便直接使用等号进行比较。
我们从链表的头节点head 开始进行遍历,遍历的指针记为pos。由于头节点一定不会被删除,因此我们可以枚举待移除节点的前驱节点,减少编写代码的复杂度。
class Solution {
public:
ListNode* removeDuplicateNodes(ListNode* head) {
if (head == nullptr || head->next == nullptr){
return head;
}
unordered_set<int> occurred = {head->val};
ListNode* pos = head;
// 枚举前驱节点
while (pos->next != nullptr) {
// 当前待删除节点
ListNode* cur = pos->next;
if (!occurred.count(cur->val)) {
occurred.insert(cur->val);
pos = pos->next;
} else {
pos->next = pos->next->next;
}
}
pos->next = nullptr;
return head;
}
};
方法二:两重循环
考虑题目描述中的「进阶」部分,是否存在一种不使用临时缓冲区(也就是方法一中的哈希表)的方法呢?
在保证方法一时间复杂度 O(N)O(N) 的前提下,是不存在这样的方法的。因此我们需要增加时间复杂度,使得我们可以仅使用常数的空间来完成本题。一种简单的方法是,我们在给定的链表上使用两重循环,其中第一重循环从链表的头节点开始,枚举一个保留的节点,这是因为我们保留的是「最开始出现的节点」。第二重循环从枚举的保留节点开始,到链表的末尾结束,将所有与保留节点相同的节点全部移除。
class Solution {
public:
ListNode* removeDuplicateNodes(ListNode* head) {
ListNode* ob = head;
while (ob != nullptr) {
ListNode* oc = ob;
while (oc->next != nullptr) {
if (oc->next->val == ob->val) {
oc->next = oc->next->next;
} else {
oc = oc->next;
}
}
ob = ob->next;
}
return head;
}
};