删除链表中等于给定值val
的所有节点。
样例
思路:处理链表问题,一般需要设置两个指针,一个指向当前节点,一个指向后继节点。然后进行删除等操作。
给出链表 1->2->3->3->4->5->3
, 和 val = 3
, 你需要返回删除3之后的链表:1->2->4->5
。
代码如下:
/**
* 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,*q = NULL;
p = head;
while(p!=NULL){
if(p->val == val){
ListNode *delNode = p;
if(p == head){
head = p->next;
q = p->next;
}
else{
q->next = p->next;
}
free(delNode);
}
else{
q = p;
}
p = p->next;
}
return head;
}
};