Scala实现移除数组中第一个负数之后的所有负数

最新推荐文章于 2025-07-13 13:33:40 发布

原创最新推荐文章于 2025-07-13 13:33:40 发布 · 516 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Scala

Scala 专栏收录该内容

108 篇文章

订阅专栏

本文介绍使用Scala语言实现从数组中移除第一个负数之后的所有负数的两种方法。第一种方法通过遍历和条件判断直接修改数组，第二种方法则优化了遍历过程并使用了中间变量保存有效索引，最后调整数组大小。这两种方法展示了不同的数组操作技巧。

package com1

import scala.collection.mutable.ArrayBuffer

/**
  * @author yangxin_ryan
  * Scala实现移除数组中第一个负数之后的所有负数
  * 版本一
  */
object Test51 {

   def main(args: Array[String]): Unit = {
      val arrBuffer = ArrayBuffer[Int]()
      arrBuffer += (1, 2, 3, 4, 5, -1, -3, -5, -7)
      var foundFirstNegative = false
      var arrLength = arrBuffer.length
      var index = 0
      while(index < arrLength) {
         if (arrBuffer(index) >= 0) {
            index += 1
         } else {
            if (!foundFirstNegative) {
               foundFirstNegative = true
               index += 1
            } else {
               arrBuffer.remove(index)
               arrLength -= 1
            }
         }
      }
      println(arrBuffer)
   }

}

package com1

import scala.collection.mutable.ArrayBuffer

/**
  * @author yangxin_ryan
  * Scala实现移除数组中第一个负数之后的所有负数
  * 版本二
  * 基于版本一优化
  */
object Test52 {

   def main(args: Array[String]): Unit = {
      val arrBuffer = ArrayBuffer[Int]()
      arrBuffer += (1, 2, 3, 4, 5, -1, -3, -5, -7)
      var foundFirstNegative = false
      val keepIndexes = for (i <- arrBuffer.indices if !foundFirstNegative || arrBuffer(i) >= 0) yield {
         print(arrBuffer(i))
         if (arrBuffer(i) < 0) foundFirstNegative = true
         i
      }
      println(keepIndexes)
      for (i <- keepIndexes.indices) {
         arrBuffer(i) = arrBuffer(keepIndexes(i))
      }
      arrBuffer.trimEnd(arrBuffer.length - keepIndexes.length)
      println(arrBuffer)
   }
}