还在用for循环遍历元素？试试for-each，它不香吗？

本文基于《Effective java》第三版第58条

在此基础之上加入了自己的理解。

A2.png

for循环是平时写代码用的最多的，但是之前看《effectivejava》大佬在某些场景写并不推荐。结合着自己之前刷算法题的经历。收益匪浅。

一、for循环的缺点

在以往遍历元素的时候，我们通常采用以下的形式：

public class Main {
    public static void main(String[] args) {
        //1、数组元素
        int[] num = new int[] {1,2,3,4,5};
        //数组的遍历
        for(int i=0;i<num.length;i++) 
            System.out.println(num[i]);
        //2、对象元素
        ArrayList<Person> lists = new ArrayList<>();
        lists.add(new Person("张三"));
        lists.add(new Person("李四"));
        lists.add(new Person("愚公要移山"));
        //对象元素的遍历
        for(Iterator<Person> it=lists.iterator();it.hasNext();) {
            Person p = it.next();
            System.out.println(p.getName());
        }
    }
}

这种写法看起来还不错，但是却并不完美。我们来分析一下，有什么缺点。然后给出解决方案。

问题1：迭代器或索引多次出现，容易造成使用错误

从上面两种遍历的代码上来看，对于数组元素是通过索引i来遍历的，但是整个for循环出现了四次i，对于对象元素是通过迭代器it来遍历的，但是整个for循环出现了三次it。在for循环遍历元素的时候，就有多次机会使用了错误的变量。而且有时候这些错误编译器无法发现。对整个应用系统造成无法预知的错误。

问题2：遍历对象元素时，需要注意容器类型

比如我们这里使用的是list，当然还有可能是其他容器类型，这些类型在更改时比较麻烦。

问题3：嵌套迭代抛出异常

这种情况比较复杂一些，先来搞个例子。比如说，我们想要列举每种花，这些花有两种属性一种是颜色，一种是大小。

public class Main {
    //枚举颜色和尺寸
    enum Color { RED, GREEN, BLUE, BLACK }
    enum Size { ONE, TWO, THREE, FOUR, FIVE,
                SIX, SEVEN, EIGHT,NINE, TEN}
    //定义花
    static class Flower{
        public Flower(Color color, Size size) {}
    }
    public static void main(String[] args) {
        Collection<Color> colors = Arrays.asList(Color.values());
        Collection<Size> sizes = Arrays.asList(Size.values());
        List<Flower> flowers = new ArrayList<>();
        //for循环添加所有的花和尺寸
        for (Iterator<Color> color = colors.iterator(); color.hasNext(); ) {
            for (Iterator<Size> size = sizes.iterator(); size.hasNext(); ) {
                flowers.add(new Flower(color.next(), size.next()));
            }   
        }   
    }
}

看似人畜无害，现在我们运行一波。

Exception in thread "main" java.util.NoSuchElementException
    at java.util.AbstractList$Itr.next(Unknown Source)
    at com.f2.Main.main(Main.java:25)

是不是感觉有点奇怪，好像双重循环遍历没啥问题，但是出现了异常，原因是外部的Color迭代器调用了多次，第一层for循环被调用了，但是又在第二层for循环内部被调用了，所以color的next被调用完了。所以出现了NoSuchElementException。但是有时候也不会出现这种情况，场景是外部循环迭代器调用的次数刚好是内部调用的n倍。

问题4：嵌套迭代不抛异常，但是结果不正确

这种情况是外部循环迭代器调用的次数刚好是内部调用的n倍。我们再来个例子：

public class Main {
    //枚举颜色
    enum Color { RED, GREEN, BLUE, BLACK }  
    public static void main(String[] args) {
        Collection<Color> colors = Arrays.asList(Color.values());
        //两层for循环
        for (Iterator<Color> c1 = colors.iterator(); c1.hasNext(); ) {
            for (Iterator<Color> c2 = colors.iterator(); c2.hasNext(); ) {
                System.out.println(c1.next()+" "+c2.next());
            }   
        }   
    }
}

现在对颜色进行for循环遍历，一共两层for循环，因为一共有四种颜色，两层for循环应该是打印16个结果。现在运行一遍看看结果：

RED RED
GREEN GREEN
BLUE BLUE
BLACK BLACK

没错，确实是打印了四条。原因和问题三是一样的。有一种方式可以很好地解决这种嵌套的问题。

嵌套迭代问题解决：

直接看代码。既然是外部的迭代器it在内部使用了，那我在内部和外部之间用一个变量缓存起来不久好了。

public class Main {
    //枚举颜色
    enum Color { RED, GREEN, BLUE, BLACK }
    public static void main(String[] args) {
        Collection<Color> colors = Arrays.asList(Color.values());
        //for循环
        for (Iterator<Color> c1 = colors.iterator(); c1.hasNext(); ) {
            //用一个变量缓存起来
            Color c = c1.next();
            for (Iterator<Color> c2 = colors.iterator(); c2.hasNext(); ) {
                System.out.println(c+" "+c2.next());
            }   
        }   
    }
}

现在再来运行，就可以很好地得出16种结果了。这种方式也比较不错，但是却不能很好地解决问题1和问题2。因此，为了解决这一现象，大佬Joshua Bloch在书中提出，推荐使用for-each循环来代替for循环。

二、for-each循环

既然作者推荐使用for-each循环，我们看看他有什么好处。是如何解决上面的问题的。

public class Main {
    //枚举颜色和尺寸
    enum Color { RED, GREEN, BLUE, BLACK }
    enum Size { ONE, TWO, THREE, FOUR, FIVE,
                SIX, SEVEN, EIGHT,NINE, TEN}
    //定义花
    static class Flower{
        public Flower(Color color, Size size) {}
    }
    public static void main(String[] args) {
        Collection<Color> colors = Arrays.asList(Color.values());
        Collection<Size> sizes = Arrays.asList(Size.values());
        List<Flower> flowers = new ArrayList<>();
        //for-each循环
        for (Color color:colors) {
            for (Size size:sizes ) {
                flowers.add(new Flower(color, size));
            }   
        }   
    }
}

看里面的for-each循环。上面的问题就全都解决了。好吧，可能你会感觉，就这？还有一个好处还没说，再往下看。

for-each 循环不仅允许遍历集合和数组，还允许遍历实现 Iterable 接口的任何对象，该接口由单个方法组成。接
口定义如下：

public interface Iterable<E> {
    // Returns an iterator over the elements in this iterable
    Iterator<E> iterator();
}

如果必须从头开始编写自己的 Iterator 实现，那么实现 Iterable 会有点棘手，但是如果你正在编写表示一组元素
的类型，那么你应该强烈考虑让它实现 Iterable 接口，甚至可以选择不让它实现 Collection 接口。这允许用户使用for-each 循环遍历类型，他们会永远感激不尽的 。

但是，有三种常见的情况是你不能分别使用 for-each 循环的:

（1）有损过滤（Destructive filtering）：如果需要遍历集合，并删除指定选元素，则需要使用显式迭代器，以便可以调用其 remove 方法。 通常可以使用在 Java 8 中添加的 Collection 类中的 removeIf 方法，来避免显式遍历。

（2）转换：如果需要遍历一个列表或数组并替换其元素的部分或全部值，那么需要列表迭代器或数组索引来替换元素的值。

（3）并行迭代：如果需要并行地遍历多个集合，那么需要显式地控制迭代器或索引变量，以便所有迭代器或索引变量都可以同步进行 。

如果发现自己处于这些情况中的任何一种，请使用传统的 for 循环，并警惕本条目中提到的陷阱 。