给出链表 1->2->3->3->4->5->3
, 和 val = 3
, 你需要返回删除3之后的链表:1->2->4->5
。
C++语言
递归形式:每次对头进行判断是否此值等于给定值val,如果相等则将此节点删除,将下一个节点作为头节点递归调用自己,否则将此节点的下一个节点为作为头节点递归调用自己,调用后的返回值作为此时头节点的下一个节点。
/*
* 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
if ( head == NULL ) {
return NULL;
} else if ( head->val ==val ) {
ListNode *newhead = removeElements(head->next, val);
free(head);
return newhead;
} else{
head->next = removeElements(head->next, val);
return head;
}
}
};
双指针:单链表无法直接访问前驱节点,需要设置2个指针pre、p,使pre是p的前节点
/**
* 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
while ( head != NULL && head->val == val){
head = head->next; //将头节点后移
}
if ( head == NULL) { //链表为空
return NULL;
}
ListNode *pre = head;
ListNode *p = head->next;
while( p != NULL) {
if ( p->val == val) {
pre->next = p->next; //将原节点p删除
} else {
pre = p; //向后移动一个节点
}
p = p->next;
}
return head;
}
};