删除链表中等于给定值 val 的所有节点。
示例:
输入: 1->2->6->3->4->5->6, val = 6 输出: 1->2->3->4->5
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* removeElements(ListNode* head, int val) {
if(!head) return head;
ListNode * pRoot=new ListNode(0);
pRoot->next=head;
ListNode* cur = head;
ListNode * pre=pRoot;
while(cur!=NULL)
{
if(cur->val ==val)
{
pre->next=cur->next;
}
else pre=cur;
cur=pre->next;
}
return pRoot->next;
}
};
这道题需要使用一个指向头节点的指针的小技巧,否则需要特别第处理头指针,处理会复杂很多。
使用一个指向头节点的指针,这样就能将这个处理过程统一起来。遍历时使用两个指针,一个指向当前节点,一个指向当前节点的前一个节点。当当前节点的值与val的值不同时,更新两个指针的值;当当前节点的值和val值相同时,删除当前节点,同时更新这两个指针。