首页 > itarticle > leetcode之203. 移除链表元素

leetcode之203. 移除链表元素

admin 11月 14, 2020 0

题目描述：

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

示例:
1
2
3
> 输入: 1->2->6->3->4->5->6, val = 6
> 输出: 1->2->3->4->5
>

解题思路：

时间复杂度：$O(n)$，空间复杂度:$O(1)$.


 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class  {
public:
    ListNode* removeElements(ListNode* head, int val) {
      
        if(!head ) return NULL;
        
         ListNode *h = new ListNode(INT32_MIN);
         h -> next = head;
         
        ListNode *p = h;
        ListNode *q = head;
     
        
        while( p -> next != NULL)
        {
           if(q -> val == val)
            {
              //ListNode *temp = q;
              //p -> next = q -> next;
              //q = p -> next;
              //delete temp;

              p -> next = q -> next;
              delete q;
              q = p -> next;
            } 
            else
            {
                p = p -> next;
                q = q -> next;  
            }
        }
        return h -> next;
    }
};