从 Set 中删除负值

Question

我预计 remove() 方法会删除负值并仅打印正整数，但输出为 -1 -2 -3 4

 import java.util.HashSet;
    import java.util.Iterator;
    import java.util.Set;

    public class GenericSetTester {
        public static void main(String[] args) {

            Set<Integer> integerSet = new HashSet<>();

            integerSet.add(0);
            integerSet.add(1);
            integerSet.add(2);
            integerSet.add(3);
            integerSet.add(4);
            integerSet.add(-1);
            integerSet.add(-2);
            integerSet.add(-3);

            for(Iterator<Integer> in = integerSet.iterator(); in.hasNext();){
                if(in.next() < 0)
                    in.remove();
                   int i = in.next();
                System.out.printf("%d ",i);
            }
        }
    }

Answer 1

@Eran 答案纠正了您当前的方法。请注意，流提供了另一种方式来处理您的要求：

Integer[] array = new Integer[] {0, 1, 2, 3, 4, -1, -2, -3};
Set<Integer> integerSet = new HashSet<>(Arrays.asList(array));
integerSet = integerSet.stream()
    .filter(i -> i >= 0)
    .collect(Collectors.toSet());

Answer 2

此循环在每次迭代中消耗Set 的两个元素（因为您在每次迭代中调用in.next() 两次），这是错误的，并导致负元素未被删除。

考虑您的Set 元素的迭代顺序：

0 // consumed by if(in.next() < 0), which returns false, element not removed
-1 // consumed by int i = in.next();, and later printed -1
1 // consumed by if(in.next() < 0), which returns false, element not removed
-2 // consumed by int i = in.next();, and later printed -2
2 // consumed by if(in.next() < 0), which returns false, element not removed
-3 // consumed by int i = in.next();, and later printed -3
3 // consumed by if(in.next() < 0), which returns false, element not removed
4 // consumed by int i = in.next();, and later printed 4

如您所见，您的循环不会从 Set 中删除任何元素，并且只打印一半元素 (-1 -2 -3 4)。

应该是：

for (Iterator<Integer> in = integerSet.iterator(); in.hasNext();) {
    int i = in.next();
    if (i < 0)
        in.remove();
    else
        System.out.printf("%d ",i);
}

正如蒂姆建议的那样，如果您在循环之后打印Set，您将看到它是否只包含它应该包含的元素。您的循环将变为：

for(Iterator<Integer> in = integerSet.iterator(); in.hasNext();){
    if(in.next() < 0)
        in.remove();
}
System.out.println (integerSet);

Answer 3

发生的情况如下： integerSet 按此顺序添加数字（但我不知道为什么）： 0:0 1：-1 2：1 3：-2 4：2 5：-3 6：3 7:4

这就是为什么如果你执行循环，它首先检查 0 的下一个是否小于 0。这将返回 true，因为 -1 小于 0。所以 0 将被删除。然后它检查 -1 的下一个是 1，小于 0。这将返回 false。所以 -1 不会被删除。

这就是为什么输出是 -1 -2 -3 4

Answer 4

一个保留原始的 lambda Set

integerSet.removeAll( integerSet.stream().filter( i -> i < 0 ).collect( Collectors.toList() ) );
…

…或简单： Holger's 解决方案 – 见下方评论

integerSet.removeIf( i -> i < 0 );