LintCode452:删除链表中的元素

来源:互联网 发布:淘宝app 商品列表 编辑:程序博客网 时间:2024/06/05 23:06

题目描述

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

测试样例

给出链表 1->2->3->3->4->5->3, 和 val = 3, 你需要返回删除3之后的链表:1->2->4->5

解题思路

这道题虽然被归为入门题,但是坑很多。上次去深信服面试就问道了这道题。

要注意两点:

1、要删除的元素可能不止一个。

2、要删除的可能是头节点。

明白了这个,就好做了。

代码如下:

/** * 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;        while (head != NULL && head->val == val) head = head->next;        ListNode *cur = head;        while (cur !=NULL && cur->next !=NULL){           if (cur->next->val == val){               cur->next = cur->next->next;           }            else               cur = cur->next;        }        return head;    }};
原创粉丝点击