Description
Remove all elements from a linked list of integers that have value val.
Example
Given: 1 –> 2 –> 6 –> 3 –> 4 –> 5 –> 6, val = 6
Return: 1 –> 2 –> 3 –> 4 –> 5
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
My solution
class Solution {
public:
ListNode *removeElements(ListNode *head, int val) {
while (head && head->val == val) head = head->next;
ListNode *cur = head;
ListNode *pre = head;
while (cur) {
if (cur->val == val) pre->next = cur->next;
else pre = cur;
cur = cur->next;
}
return head;
}
};
上述代码基本思路是分成两个指针, pre走过的都是新链表生成的node, cur是探索指针, 如果cur->val==val就修改pre->next, 并且cur一直走下去. 这么写思路还算清晰, 但是双指针在这个问题上略显臃肿.
为同样实现pre的作用, 实际上在判断的时候, 不判断node->val, 取而代之的是判断node->next->val, 跟我上面代码区别变成: pre=cur替换成cur前进, 否则就是cur不动, next前进!
Discuss
有神代码如下, 简洁的递归实现:
public ListNode removeElements(ListNode head, int val) {
if (head == null) return null;
head.next = removeElements(head.next, val);
return head.val == val ? head.next : head;
}
分析:
递归的核心在于分解问题, 子问题求解, 以及子问题合并!
回想几种递归的实现:
1. 斐波那契
- 分解问题: f(大数)是f(小数)之和
- 子问题求解: 最底层的递归是根据初值做加法
- 合并子问题: 直接相加!!
2. 归并排序
- 分解问题: 长序列排序分解成许多小序列的排序
- 子问题求解: a. 2个数, 比较大小即可 b. 1个数, 已排序
- 子问题合并: 对两个已排序的序列合并为一个排序序列, 有高效的算法方式
3. 本文问题
- 分解问题: 长链表由短链表顺序组成
- 子问题求解: 是啥就返回啥, 除非不满足val这个筛选
- 子问题合并: 链表直接拼接