LintCode 删除链表中等于给定值val的所有节点

给出链表 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;
    }
};

你可能感兴趣的:(算法刷题)