LeetCode刷题实战82:删除排序链表中的重复元素 II
算法的重要性,我就不多说了吧,想去大厂,就必须要经过基础知识和业务逻辑面试+算法面试。所以,为了提高大家的算法能力,这个公众号后续每天带大家做一道算法题,题目就从LeetCode上面选 !
今天和大家聊的问题叫做 删除排序链表中的重复元素 II,我们先来看题面:
https://leetcode-cn.com/problems/remove-duplicates-from-sorted-list-ii/
Given a sorted linked list, delete all nodes that have duplicate numbers, leaving only distinct numbers from the original list.
题意
示例 1:
输入: 1->2->3->3->4->4->5
输出: 1->2->5
示例 2:
输入: 1->1->1->2->3
输出: 2->3
解题
解法
class Solution {
public:
ListNode* deleteDuplicates(ListNode* head) {
// 如果元素少于2个则直接返回
if (head == nullptr || head->next == nullptr) return head;
int cur = head->val;
// 初始化
bool flag = false;
ListNode* ct = new ListNode(0);
ListNode* pnt = ct;
head = head->next;
// 遍历元素
while (head) {
int hd = head->val;
// 判断当前元素与之前的元素是否相等
if (hd != cur) {
// 之前的元素没有出现重复
if (!flag) {
// 把之前的元素存入链表
pnt->next = new ListNode(cur);
// 链表移动
pnt = pnt->next;
}else flag = false;
// 更新之前的元素
cur = hd;
}else flag = true;
head = head->next;
}
// 单独处理最后一个元素
if (!flag) pnt->next = new ListNode(cur);
return ct->next;
}
};
class Solution:
def deleteDuplicates(self, head: ListNode) -> ListNode:
# 创建一个新的元素放在链表头部,这样原本第一个元素就是头指针的下一个元素了
node = ListNode()
node.next = head
# pnt指向新创建的元素
pnt = node
while pnt.next is not None:
# cur指向当前位置的下一个位置
# 我们要做的即判断cur这个指针的元素是否重复
cur = pnt.next
if cur.next is None:
break
# ptr指向cur的next
ptr = cur.next
# 如果ptr和cur相等,那么出现重复,我们需要跳过所有相等的元素
if ptr is not None and ptr.val == cur.val:
while ptr is not None and ptr.val == cur.val:
ptr = ptr.next
pnt.next = ptr
# 否则说明不重复,移动pnt
else:
pnt = pnt.next
# 由于我们开始的时候人为添加了一个辅助元素
# 返回的时候要将它去除
return node.next
总结
上期推文: