c_cpp 203.删除链接列表元素 - easy - 2018.8.1

Posted 2021-05-14

/**
1：先把链表头部那些不合法的部分给干掉
2：set preNode 和 curNode，如果 curNode 需要干掉，preNode 就不动，否则都往后面移动
 */
class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {
        while (head && head->val == val) {
            head = head->next;
        }
        if (head == NULL) return NULL;
        ListNode *preNode = head;
        ListNode *curNode = head->next;
        while (curNode != NULL) {
            if (curNode->val == val) {
                preNode->next = curNode->next;
            } else {
                preNode = curNode;
            }
            curNode = curNode->next;
        }
        return head;
    }
};