为什么我不能删除链表中的匹配项？

Question

问题：给定一个值，从链表中删除该值的所有实例。 更多信息如下： JAVA

    /**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
public class Solution {
    public ListNode removeElements(ListNode head, int val) {
        ListNode n = head; //1, 2, 6, 3, 4, 5, 6
        while(n.next == null){
            if(n.next.val == val){
                n.next = n.next.next;
            }
            n = n.next;
            if(n == null){break;}
        }
        return head;
    }
}

既然它是通过引用传递的，它应该更新不应该吗？

我试过了：

removeElements([1,2,6,3,4,5,6], 6)

但它没有删除任何东西。那么我做错了什么？

Answer 1

有几个问题：

• 你想循环直到一个节点为空，直到不为空（即while( ... != null)）
• 您可能希望循环直到n 为空，直到n.next 为空，否则您将跳过最后一个元素
• 您要检查 n.val == val 而不是 n.next.val == val，否则您会跳过第一个元素
• 如果您检查n，您希望跟踪前一个节点，以防您需要删除n，即prev.next = n.next。
• 如果要删除第一个元素（头部），则需要替换头部，即返回第二个元素（这可以通过检查prev == null 来完成，这意味着n 是当前头部）。

Answer 2

正如 Thomas 在他的第一点中提到的，您错误地编写了 while 循环。此外，因为您有一个单链表，所以您需要跟踪前一个节点（Thomas 也提到过）。

public static ListNode removeElements(ListNode head, int val)
{
    ListNode n = head;
    ListNode prev = null; //the previous node.
    while (n != null)
    {
        if (n.value == val)
        {
            //if prev is null it means that we have to delete the head
            //and therefore we need to advance the head
            if (prev == null)
            {
                head = n.next;
                prev = null;//remains at null because there's nothing before head
                n = head;//new head, let's start from here
            } else
            {
                prev.next = n.next; //let's skip the element to delete
                n = n.next;//let's advance to the next node
            }
        } else
        {
            //nothing to delete, let's advance to the next node and make 
            //the current node the previous node in the next iteration
            prev = n;
            n = n.next;
        }

    }
    return head;
}

Answer 3

使用适当的测试用例解决这些问题始终是一个好习惯。设计可能的测试用例，包括角落用例。然后按照你的算法，看看它是否解决了测试用例的问题。编写代码并试运行它，这将对代码和逻辑进行完整性检查。

以下是该问题必须编写测试用例的案例。

1. 空列表，
2. 一个元素且值等于要删除的值的列表
3. 一个元素且值不等于要删除的值的列表
4. 具有两个元素且第一个节点值等于要删除的值的列表
5. 具有两个元素且最后一个节点值等于要删除的值的列表
6. 具有三个元素且第一个节点值等于要删除的值的列表
7. 具有三个元素且第二个节点值等于要删除的值的列表
8. 具有三个元素且最后一个节点值等于要删除的值的列表

下面是删除链表中给定值节点问题的代码

public ListNode removeElements(ListNode head, int val) {
        while(head != null && head.val == val){
            head = head.next;
        }
        if(head == null){
            return null;
        }
        ListNode node = head;
        while(node.next != null){
            if(node.next.val == val){
                node.next = node.next.next;
            }else{
                node = node.next;
            }
        }
        return head;
}