[LintCode]452.删除链表中的元素

论坛 期权论坛 脚本     
已经匿名di用户   2022-5-29 19:04   707   0

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


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


思路:1.判断head是否为空

2.要删除的节点在头部

3.要删除的节点在中间、尾部,遍历链表删除

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



/**
 * 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) {
        if(head==NULL)
            return head;
        ListNode *p=head;
        ListNode *newHead=new ListNode(0);
        ListNode *node=newHead;
        while(p!=NULL){
            if(p->val!=val)
            {
                node->next=p;
                node=node->next;
            }
            p=p->next;
        }
        node->next=NULL;
        return newHead->next;
    }
};



分享到 :
0 人收藏
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

积分:81
帖子:4969
精华:0
期权论坛 期权论坛
发布
内容

下载期权论坛手机APP