码迷,mamicode.com
首页 > 其他好文 > 详细

[LeetCode笔记]-203-Remove Linked List Elements

时间:2015-09-26 21:20:03      阅读:237      评论:0      收藏:0      [点我收藏+]

标签:

删除链表中指定数值的元素,考虑头指针、返回。

/**
 * 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) {
        ListNode* newHead = new ListNode(36);
        newHead->next = head;
        ListNode* pre =  newHead;
        ListNode* cur =  head;
        while(cur!=NULL)
        {
            if(cur->val == val)
            {
                cur = cur->next;
                pre->next = cur;
            }
            else
            {
                cur = cur->next;
                pre = pre->next;
            }
        }
        return newHead->next;
    }
};

 一开始没有想过用递归写,Discussion提供了如下写法,很简洁。先递归到链表尾部,再逐层返回。

/**
 * 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==NULL) 
           return NULL;
        head->next = removeElements(head->next, val);
        return head->val == val ? head->next : head;
    }
};

 

[LeetCode笔记]-203-Remove Linked List Elements

标签:

原文地址:http://www.cnblogs.com/xajhzdfbb/p/4841235.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!