scala forif的例子

scala forif的例子

package demo.forif



object ForIfDemo {

  /**

    * 将Vector中的大于5的偶数添加到新的Vector中并返回。

    * @param v

    * @return

    */

  def evenGT5(v:Vector[Int]):Vector[Int] = {

    var result = Vector[Int]()

    for {

      n <- v

      if n > 5

      if n % 2 == 0

    }

    result = result :+ n

    result

  }



  /**

    * 将Vector中小于10的所有奇数都添加到新的vector中并返回。

    * @param v

    * @return

    */

  def yielding(v: Vector[Int]):Vector[Int] = {

    val result = for {

      n <- v

      if n < 10

      if n % 2 != 0

    } yield n       //for循环语句后面跟上yield n会自动把n记录到Vector中最后返回Vector。

    return result

  }



  def main(args: Array[String]): Unit = {

    val v = Vector(1,2,3,5,6,7,8,10,13,14,17)

    val newV = evenGT5(v)

    newV.foreach(n => print(String.valueOf(n) + "\t"))

    println()

    val newV2 = yielding(v)

    newV2.foreach(n => print(String.valueOf(n) + "\t"))

  }

}

运行结果：

6    8    10    14    
1    3    5    7