leetcodeのLinkedList

LinkedList

来到了leetcode的链表专题。

203. 移除链表元素

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

示例:

输入: 1->2->6->3->4->5->6, val = 6
输出: 1->2->3->4->5

题虽然不难，但是用到了链表中一个非常重要的思想虚拟头节点思想，就是链表头节点之前加一个空节点来当头节点 ，原来的头节点就变成第2个节点了。之所以这样做是为了操作方便，如果没有虚拟头节点的话，在对链表操作时要对真正的头节点写专门的处理代码。而有了虚拟头节点后专门的处理代码就负责虚拟头节点 但是因为虚拟头节点并不参与具体的事务，专门的处理代码就可以省略了。

class Solution {
    public ListNode removeElements(ListNode head, int val) {
        ListNode vhead = new ListNode(-1,head);
        ListNode pre = vhead;
        ListNode cur = vhead.next;        
        while(cur != null){
            if(cur.val == val){
                pre.next = cur.next;
                cur = pre.next;
            }else{
                pre = pre.next;
                cur = cur.next;
            }
        }
        return vhead.next;
    }
}

如果不加虚拟头节点 ，代码就要复杂一些了。

class Solution {
    public ListNode removeElements(ListNode head, int val) {
        //删除值相同的头结点后，可能新的头结点也值相等，用循环解决
        while(head!=null&&head.val==val){
            head=head.next;
        }
        if(head==null)
            return head;
        ListNode prev=head;
        //确保当前结点后还有结点
        while(prev.next!=null){
            if(prev.next.val==val){
                prev.next=prev.next.next;
            }else{
                prev=prev.next;
            }
        }
        return head;
    }
}

206. 反转链表

反转一个单链表。

示例:

输入: 1->2->3->4->5->NULL
输出: 5->4->3->2->1->NULL

题目的要求非常简单，在这题中用到了虚拟头节点的思想，不过这里的虚拟头节点并不是一个节点而是一个null 。

先来看看最开始我写错❌的版本

class Solution {
    public ListNode reverseList(ListNode head) {
        ListNode pre = null;
        ListNode cur = head;
        ListNode next = cur.next;
        while(cur != null){
            cur.next = pre;
            pre = cur;
            cur = next;
            next = next.next; // 问题代码
        }
        return pre;
    }
}

问题出在最后一个节点时，cur指针指向最后一个节点 ， next指针指向null ，那么next.next就会发生java.lang.NullPointerException了。

来看看正确的写法

class Solution {
    public ListNode reverseList(ListNode head) {
        ListNode pre = null;
        ListNode cur = head;       
        while(cur != null){
            ListNode next = cur.next;
            cur.next = pre;
            pre = cur;
            cur = next;
        }
        return pre;
    }
}

这样写的妙处在于有个当前指针不为空的判断，只有cur不为空才创建next指针 。保证了在cur指向最后一个节点时， next是null当下一轮循环时cur为空，也就不存在next指针了。