将循环语句改为for时有问题

Question

老师，请问能将虚拟头结点的循环语句改为for循环吗？
for(ListNode prev = dummyHead; prev.next != null ; prev = prev.next){
if(prev.next.val == val){
ListNode delNode = prev.next;
prev.next = delNode.next;
delNode.next = null;
}
}
提交到Leetcode上会报错，我自己用IDEA测试了，当链表结尾为6的时候[1,2,3,6,5,6]，会报错。其他结尾时就不会。

Answer 1

liuyubobobo 2022-08-28 23:30:26

你的这个 for 循环的逻辑和我课程中的 while 循环逻辑不等价。

课程中的这个 while 循环：

while(prev.next != null){
    if(prev.next.val == val)
        prev.next = prev.next.next;
    else
        prev = prev.next;
}

在 prev.next.val == val 的时候，做了删除操作，但是 prev 没有往后移；

只有在 prev.next.val != val 的时候，prev 才后移（执行 prev = prev.next）

但是你的 for 循环，无论什么情况，prev 都后移。所以是错误的。想想为什么？

改成这样即可：

class Solution {
    public ListNode removeElements(ListNode head, int val) {
        
        ListNode dummyHead = new ListNode(-1);
        dummyHead.next = head;
        
        for(ListNode prev = dummyHead; prev.next != null ; ){
            if(prev.next.val == val){
                ListNode delNode = prev.next;
                prev.next = delNode.next;
                delNode.next = null;
            }
            else
                prev = prev.next;
        }
        
        return dummyHead.next;
    }
}

继续加油！：）

• 提问者 XLL_XOI #1

  谢谢老师