LinkedList 的迭代器类的 remove()

Question

所以我试图更好地理解 LinkedLists，一个练习告诉我为我编写的链表类添加实现迭代器类的 remove() 方法。 我的迭代器类如下所示：

public java.util.Iterator<T> iterator() {
    return new java.util.Iterator<T>() {
        Node prev= null,curr = head;

        public boolean hasNext() {  
            if (curr != null) {
                return true;
            }
            return false;
        }

        public T next() {
            T temp = curr.data;
            prev = curr;
            curr = curr.next;
            return temp;
        }

        public void remove() {
            if(prev==null || curr==null)
                head=head.next;
            else
                prev.next=curr.next;
        }
    };
}

我为它写的一个测试有点像这样：

public void testiterator(){
    BasicLinkedList<String> basicList = new BasicLinkedList<String>();
    basicList.addToFront("Blue").addToEnd("Red").addToFront("Yellow");
    for(Iterator<String> i = basicList.iterator(); i.hasNext();){
        if(i.next().equals("Blue"))
            i.remove();
    }
    assertTrue(basicList.toString().equals("\" Yellow Red \""));
}

但是，当我打印 basicList 时，它告诉我列表包含黄色和蓝色而不是黄色和红色。我是在执行 remove() 方法错误，是我使用错误，还是两者兼而有之？

感谢你们的时间！

Answer 1

问题在于curr 不是指返回的最后一个元素，而是指要返回的下一个元素。

remove() 是要删除前者，而您的方法是删除后者。

Answer 2

为什么不直接将当前节点设置为下一个节点。为什么要在 remove 方法中检查 null。

Answer 3

试试这个伙伴：

public java.util.Iterator<T> iterator() {
    return new java.util.Iterator<T>() {
        Node<T> prev = null;
        Node<T> curr = null;

        public boolean hasNext() {
            if (curr == null) {
                return (head != null);
            }
            return (curr.next != null);
        }

        public T next() {
            if (!hasNext()) {
                return null;
            }
            if (curr == null) {
                curr = head;
            } else {
                prev = curr;
                curr = curr.next;
            }
            return curr.data;
        }

        public void remove() {
            if (curr != null) {
                if (prev != null) {
                    prev.next = curr.next;
                } else {
                    head = curr.next;
                }
            }
        }
    };
}