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

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

一、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 循环,并警惕本条目中提到的陷阱

相关文章
|
3月前
|
算法 索引 Python
【Leetcode刷题Python】34. 在排序数组中查找元素的第一个和最后一个位置(二分查找)
解决LeetCode "在排序数组中查找元素的第一个和最后一个位置" 问题的方法。第一种方法是使用两次二分查找,首先找到目标值的最左边界,然后找到最右边界。第二种方法是利用Python的list.index()方法,先正序找到起始位置,再逆序找到结束位置,并给出了两种方法的Python实现代码。
61 0
|
存储 JavaScript 前端开发
手撕前端面试题【javascript~ 列表动态渲染、无重复数组、数组排序、新数组、创建数组、深浅拷贝、内存泄露等】
html页面的骨架,相当于人的骨头,只有骨头是不是看着有点瘆人,只有HTML也是如此。 css,相当于把骨架修饰起来,相当于人的皮肉。
77 0
|
Go 索引 Cloud Native
【刷题日记】34. 在排序数组中查找元素的第一个和最后一个位置
本次刷题日记的第 72 篇,力扣题为:34. 在排序数组中查找元素的第一个和最后一个位置 ,中等
|
6月前
|
测试技术
【刷题-牛客】出栈、入栈的顺序匹配 (代码+动态演示)
【刷题-牛客】出栈、入栈的顺序匹配 (代码+动态演示)
107 0
|
算法
代码随想录训练营Day1:二分查找与移除元素
代码随想录训练营Day1:二分查找与移除元素
44 0
|
算法
【算法挨揍日记】day10——704. 二分查找、34. 在排序数组中查找元素的第一个和最后一个位置
给定一个 n 个元素有序的(升序)整型数组 nums 和一个目标值 target ,写一个函数搜索 nums 中的 target,如果目标值存在返回下标,否则返回 -1。
363 0
|
Python
while循环的妙用
while循环的妙用
89 1
|
数据库管理 Python
巧用for循环
巧用for循环
60 0
List 集合遍历过程中删除元素。这个坑踩一遍就不要再踩了!
List 集合遍历过程中删除元素。这个坑踩一遍就不要再踩了!
|
算法 前端开发 索引
有序的数组,试试用指针法遍历
有序的数组,试试用指针法遍历
77 0