我们对给定的链表进行一次遍历,并用一个哈希集合(HashSet)来存储所有出现过的节点。由于在大部分语言中,对给定的链表元素直接进行「相等」比较,实际上是对两个链表元素的地址(而不是值)进行比较。因此,我们在哈希集合中存储链表元素的值,方便直接使用等号进行比较。
具体地,我们从链表的头节点 \textit{head}head 开始进行遍历,遍历的指针记为 \textit{pos}pos。由于头节点一定不会被删除,因此我们可以枚举待移除节点的前驱节点,减少编写代码的复杂度。
这样枚举有什么好处?试想一下,如果我们直接枚举待移除节点,那么在将它进行移除时,我们本质上是将它的前驱节点连向后继节点。而由于题目给定的链表结构中,我们无法直接访问一个节点的前驱节点。因此,我们不如直接枚举前驱节点 u,那么节点本身就是 u.next,后继节点就是 u.next.next。
在遍历完成后,我们就得到了最终的答案链表。
struct ListNode* removeDuplicateNodes(struct ListNode* head) {
if (head == NULL) {
return head;
}
int* occurred = (int*)calloc(20001, sizeof(int));
occurred[head->val] = 1;
struct ListNode* pos = head;
// 枚举前驱节点
while (pos->next != NULL) {
// 当前待删除节点
struct ListNode* cur = pos->next;
if (!occurred[cur->val]) {
occurred[cur->val] = 1;
pos = pos->next;
} else {
pos->next = pos->next->next;
}
}
pos->next = NULL;
return head;
}