删除链表中等于给定值val
的所有节点。
样例
给出链表 1->2->3->3->4->5->3
, 和 val = 3
, 你需要返回删除3之后的链表:1->2->4->5
。
解题思路:对于元素特判一下是否为首个,如果是这返回head-》next,否则保存头指针,遍历找到和val 相等的,将指向它的指向下一个。
class Solution {
public:
/**
* @param head a ListNode
* @param val an integer
* @return a ListNode
*/
ListNode *removeElements(ListNode *head, int val)
{
// Write your code here
int i=0;
ListNode *a;
a=head;
if(head==NULL)
return NULL;
while(head->next!=NULL)
{
if(head->next->val==val)
{
if(head->next->next==NULL)
{
head->next=NULL;
break;
}
else
{
head->next=head->next->next;
}
}
else
head=head->next;
}
if(a->val==val)
return a->next;
return a;
}
};
删除链表中等于给定值val
的所有节点。
样例
给出链表 1->2->3->3->4->5->3
, 和 val = 3
, 你需要返回删除3之后的链表:1->2->4->5
。