您的位置:首页 > 其它

[lintcode]Remove Linked List Element 删除链表中的元素

2017-03-31 10:29 471 查看
指定一个值,移除所有链表上是这个值的节点。需要注意头指针也有值,注意头指针为空的情况。

/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
/**
* @param head a ListNode
* @param val an integer
* @return a ListNode
*/
ListNode *removeElements(ListNode *head, int val) {
// Write your code here
ListNode *p;
ListNode *q;

while(head!=NULL&&head->val == val){
head = head->next;
}

if(head == NULL)
return head;

p = head;
q = head->next;
while (q!=NULL){
if(q->val == val){
p->next = q->next;
delete q;
q = p->next;
}
else{
p = p -> next;
q = q -> next;
}
}
return head;
}
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  lintcode leetcode