LeetCode203. 移除链表元素

链表的“删除”,往往采用的是跳过的方式,即如果p -> next是我们要“删除”的节点,我们只要把p的next指针指向p -> next -> next就可以了,也就是让链表跳过p -> next,让这个节点“社会性死亡”。

根据这个思路,这一题里我们只需要遍历链表,判断节点值是否与我们要删除的相等,如果相等,就跳过这个链表。

不过考虑到有可能链表一开始就是我们要删除的节点,所以我们需要先找到第一个不为要删除的值V的节点,将这个节点作为head,然后从这个节点开始遍历到链表结尾,删除节点。

代码如下:

/**
 * 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 V) {
        if(head == NULL) {
            return NULL;
        }
        while(head != NULL && head -> val == V) {            //由于链表的开头元素就有可能需要删除,而这时我们还没有节点有next指针指向这个元素呢,因此需要先找到我们的链表头,也就是第一个值不为V的节点
            head = head -> next;
        }
        ListNode* p = head;                        //确定链表头之后,就可以开始遍历链表,根据节点的next指针判断每个节点是否要删除了
        while(p != NULL) {
            if(p -> next != NULL && p -> next -> val == V) {      //删除的情况就是,p的下一个节点不为空,且下一个节点值就是我们要删除的值
                p -> next = p -> next -> next;
            } else {                              //继续往后找
                p = p -> next;
            }
        }
        return head;
    }
};
posted @ 2020-08-04 18:10  machine_gun_lin  阅读(61)  评论(0编辑  收藏  举报