菜鸡大学生的数据结构——刷题篇1
自从宇宙大爆炸以来,上课就和课后作业联系起来。大学生不能失去刷题,正如西方不能失去耶路撒冷。 我们的菜鸡大学生也知道这一点,虽然内心有一万个不愿意,但是还是打开了leetcode,破防了三天三夜,编出了这篇文章。
敬请见证…
移除链表元素
给你一个链表的头节点 head
和一个整数 val
,请你删除链表中所有满足 Node.val == val
的节点,并返回 新的头节点 。
示例1:
输入: head = [1,2,6,3,4,5,6], val = 6
输出: [1,2,3,4,5]
示例2:
输入: head = [], val = 1
输出: []
示例3:
输入: head = [7,7,7,7], val = 7
输出: []
分析
无论在什么时候,题目的示例也都是很重要的,它可以给我们考虑到大部分情况,剩下的一小部分则需要我们去思考。
这一题就是这样,通过题目我们可以发现有三种情况:
- 最常见最普通的情况。
- 空链表。
- 链表里面全是要移除的数据。
然后根据我们多年做题的经验,怎么能没有要删除的数据连续出现的情况呢,果断加上,至于以后用不用得到再说。
第一种情况
根据我们写链表的经验,这个不就是pos位置删除数据吗?只不过pos位置变成了满足条件的值。
直接来吧!
逻辑:next,cur和prev三个个指针遍历链表,如果cur的值是要删除的值,prev指向cur的下一位,释放cur,cur指向下一位,直到遍历结束。
我们试着把这套逻辑扔到要删除的数据连续出现这一种情况。发现是可行的(可以自己试着画图验证),于是,我们轻松愉快的解决掉了两种情况。
第二种情况
空链表欸!是空链表欸!
直接返回NULL就好了欸!
我的老天鹅啊太快乐了。
第三种情况
我们总结的其实是第三种情况的过激版本,其实真正的第三种情况是开头是要删除的数据该怎么办。这种情况会让prev被打爆,变成一个永远瞎指的有害的废物。
想解决它也不难,如果开头是要删除的数字val,我们就把开头删掉,直到开头不是val之后再执行第一种情况的逻辑就好。
好,我们所有情况都解决了,接下来上代码:
代码
struct ListNode* removeElements(struct ListNode* head, int val){
if(head==NULL)
return head;
while(head&&