LintCode 452.删除链表中的元素 C++

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

问题描述:删除链表中等于给定值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) {}
};
分析:首先要判断链表是否为空,是的话直接返回,然后考虑到单链表的删除需要两个指针变量的配合作用,那么就要来定义两个变量,一个 cur 保存当前节点,另一个 nex 保存当前节点的下一个节点,从链表的第二个节点开始遍历,若 nex 的值与 val 的值相等,则将 cur->next 指向 cur->next->next( nex->next) ,删除节点。继续移动将 cur nex 下移一位,往后遍历,直到链尾 (nex 为空 ) 。再判断头结点 head value 是否与 val 相等,如果相等,将头结点下移,删除原头结点,否则不做任何操作。遍历结束后返回 head

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;  
        }
        ListNode *cur = head;  
        ListNode *nex = head -> next;  
        while (nex != NULL){  
            if(nex -> val == val){  
                cur -> next = nex -> next;  
                nex = nex -> next;  
            }else{  
                cur = cur -> next;  
                nex = nex -> next;  
            }  
        } 
        if(head -> val == val) head = head -> next;  
        return head;  
    }
};
注意:要解决该问题,首先要熟悉单链表节点的删除,我的第一反应时保存当前节点和上一节点,单链表中对上一节点的定位有些麻烦,本例中定义当前节点和下一节点的方法效率更高一些。

其次,对于头结点的判断,我刚开始觉得放在判断头结点是否为空之后也可以,试了一下发现报错了,例如原链表1->1->null和val为1,期望值null,结果1->null。出现的问题是,本例从第二个节点开始遍历,除了第一个节点外已经没有值与val相等的节点了,而提前判断之后再遍历,则会少判断第二个节点的值。

个人看法,如有错误,请批评指出。



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

本版积分规则

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

下载期权论坛手机APP