[Leetcode]移除链表元素

题目#

代码 #

/**
 * 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 val) {
          if(head==nullptr)
           return nullptr;
          //头结点为要删除的结点
          auto ptr=head;
          bool isHeadDelete=false;
          ListNode* pre=nullptr;
          while(ptr!=nullptr)
          {
              //找到
              if(ptr->val==val)
              {
                  //说明为头结点
                  if(pre==nullptr)
                  {
                      auto del=ptr;
                      ptr=ptr->next;
                      delete del;
                      head=ptr;
                      isHeadDelete=true;
                  }
                  else
                  {
                      auto del=ptr;
                      ptr=ptr->next;
                      pre->next=ptr;
                      delete del;
                  }
              }
              else//没有找到
              {

                  pre=ptr;
                  ptr=ptr->next;
              }
              if(isHeadDelete)
              {
                  isHeadDelete=false;
                  pre=nullptr;
              }

          }
          if(head!=nullptr)
                return head;
          return nullptr;
    }
};