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
题意清楚,主要考察链表指针的使用。
小心头结点和尾节点的处理。
时间36ms。
代码:
/**
* 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 == NULL)//空链表
return head;
ListNode *p, *q;
p=q=head;
while( q!=NULL && q->val == val )//确定头结点
{
q=p->next;
head = p = q;
}
if( q == NULL)
return head;
q=p->next;
while( q!=NULL )
{
if(q->val == val)//如果出现val
{
q=q->next;
}
else
{
p->next = q;
p=q;
q = q->next;
}
}
p->next = q;
return head;
}
};