Scala：使用数组中的值在另一个数组中搜索

Question

我有两个数组:

globalaray:Array（Int，Array[String]）和SpecificArray:Array（Int，Int）。

它们中的第一个Int是一个键，我想从GlobalArray中获取与该键对应的元素。

在伪代码中:

val v1
For each element of SpecificArray
    Get the corresponding element from GlobalArray to use its Array[String]
         If (sum <= 100)
             for each String of the Array
                 update v1 
                 // ... some calculation
                 sum += 1

println (v1)

我知道使用＜code＞。map（）我可以遍历＜code＞SpecificArray

SpecificArray.map{x => val in_global = GlobalArray.filter(e => (e._1 == x._1))
                       // I don't know how to follow
                  }

Answer 1

sum如何影响逻辑以及v1到底是什么，从您的代码中尚不清楚，但看起来您多次搜索GlobalArray。如果是这样，则将此数组转换为更易于搜索的数据结构是有意义的：Map。你可以这样做

val globalMap = GlobalArray.toMap

然后你可以像这样连接琴弦

println(SpecificArray.flatMap({case (k,v) => globalMap(k).map(s => (k,v,s))}).toList)

如果您只需要字符串，则可以使用

println(SpecificArray.flatMap({case (k,v) => globalMap(k)}).toList)

请注意，此代码假定对于SpecificArray中的每个键，GlobalArray中都会有一个匹配的键。如果不是这种情况，您应该使用其他方法访问Map，如getOrElse：

println(SpecificArray.flatMap({case (k,v) => globalMap.getOrElse(k, Array()).map(s => (k,v,s))}).toList)

更新

如果 sum 实际上是 count，并且它适用于整个“联接”数据，而不是 SpecificArray 中的每个键，则可以使用 take 而不是它。代码将如下所示：

val joined = SpecificArray2.flatMap({case (k,v) => globalMap.getOrElse(k, Array()).map(s => (s,v))})
        .take(100) // use take instead of sum

然后，您可以以任何您想要的方式使用加入。更新的演示将 v1 构建为形式 v1 = String_of_GlobalArray “ = ” 的连接字符串 2nd_Int_of_SpecificArray就在这里。这个想法是使用mkString而不是显式变量更新。

Answer 2

笔记需要更多优化

将GlobalArray转换为< code>Map以加快查找速度。

val GlobalMap = GlobalArray.toMap

SpecificArray.flatMap(x => GlobalMap(x._1))
.foldLeft(0)((sum:Int, s:String) => {
    if(sum<=100) {
    // update v1
    // some calculation
    }
    sum+1
})

如果不是 GlobalMap 中存在 SpecificArray 的所有键，请使用 GlobalMap.getOrElse（x._1， Array（））

Answer 3

下面这样的怎么样，我更喜欢for理解代码，它有更好的易读性。

var sum:Int = 0
var result:String = ""
for {
  (k1,v1) <- SpecificArray //v1 is the second int from specific array
  (k2,values) <- GlobalArray if k1 == k2  //values is the second array from global array
  value  <- values if sum < 100  //value is one value frome the values
   _ = {sum+=1; result += s"(${v1}=${value})"} //Update something here with the v1 and value
 } yield ()

println(result)