LintCode【入门】删除链表中的元素

原创于 2016-10-16 16:38:26 发布 · 238 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#C++ #LintCode

LintCode 专栏收录该内容

5 篇文章

订阅专栏

本文介绍了一种算法，用于从单链表中移除所有值为特定整数的节点，并提供了一个具体的示例：给定链表1->2->3->3->4->5->3，在删除所有值为3的节点后，链表变为1->2->4->5。

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

样例

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

/**
 * 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 head;
        while (head->val == val) {
            head = head->next;
            if (head == NULL) return head;
        }
        ListNode *p, *q;
        p = head;
        q = p->next;
        while (q!=NULL) {
            if (q->val == val) {
                q = q->next;
                p->next = q;
            } else {
                p = p->next;
                q = q->next;
            }
        }
        return head;
    }
};