LeetCode two sum -- scala 题解与思路

Posted shiter

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了LeetCode two sum -- scala 题解与思路相关的知识,希望对你有一定的参考价值。


题目

给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。

你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。

你可以按任意顺序返回答案。

示例 1:

输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
示例 2:

输入:nums = [3,2,4], target = 6
输出:[1,2]
示例 3:

输入:nums = [3,3], target = 6
输出:[0,1]

提示:

2 <= nums.length <= 104
-109 <= nums[i] <= 109
-109 <= target <= 109
只会存在一个有效答案
进阶:你可以想出一个时间复杂度小于 O(n2) 的算法吗?

吐槽一下 LeetCode 的scala 没有关键词提示,还运行慢


题解 1 ---- 差集数组

用target 把 原来的数组减去,然后遍历两个数组,对应下标相加 = target 输出第一个数组的下标

错误解法

import scala.collection.mutable.ArrayBuffer

object TwoSum 

  def twoSum(nums: Array[Int], target: Int): Array[Int] = 

    var temp_nums:Array[Int] = new Array[Int](nums.length)

    for ( i <- 0 to (temp_nums.length - 1)) 
      temp_nums(i) = target-nums(i)
    

    var result:ArrayBuffer[Int] = new ArrayBuffer[Int]()
    for ( i <- 0 to (nums.length - 1)) 
      for(j <- 0 to (temp_nums.length - 1))
        if(nums(i)==temp_nums(j))
          result +=  i
        
      
    
    result.toArray
  


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

    val nums = Array(7,1,2,3,4)
    val target = 8
    val result = twoSum(nums, target)

    println(result)
  





限制条件 与 结束条件

对比数组的下标位置不能一样,比如 4,这种特殊情况上下一样代表它正好是target 的一半

上述代码输出的时候还有一个问题,是没有立即结束,当遍历到1 的时候有发现了一次结果,覆盖了上一次正确结果,正确的应该是有结果了立即返回

修改后通过测试



object TwoSum 

  def twoSum(nums: Array[Int], target: Int): Array[Int] = 

    var temp_nums:Array[Int] = new Array[Int](nums.length)

    for ( i <- 0 to (temp_nums.length - 1)) 
      temp_nums(i) = target-nums(i)
    

    var result:Array[Int] = new Array[Int](2)
    for ( i <- 0 to (nums.length - 1)) 
      for(j <- 0 to (temp_nums.length - 1))
        if((nums(i)==temp_nums(j))&&(i!=j))
          result(0) = i
          result(1) = j
          return result
        
      
    
    result
  


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

//    val nums = Array(7,1,2,3,4)
//    val target = 8
      val nums = Array(2,7,11,15)
      val target = 9
      val result = twoSum(nums, target)

   result.foreach(x=>println(x))
  



跑个这,占了51 m 内存,也是够绝的。。。


题解 2 ---- 两层循环语法糖版本


object Solution 
    def twoSum(nums: Array[Int], target: Int): Array[Int] = 
    
    for(i<- nums.indices;j<-i+1 until nums.length)
        if(nums(i)+nums(j)==target)
            return Array(i,j)
            
        
        null
    


题解 3 ---- hash 表

object Solution 
    def twoSum(nums: Array[Int], target: Int): Array[Int] = 
        
        var tempMap:Map[Int, Int] = Map()
        
        for(i<-nums.indices)
            if(tempMap.contains(nums(i)))
                return Array(tempMap(nums(i)),i)
            
            tempMap+=((target - nums(i))->i)
        
        
        Array(0,0)
    


题解 4 ---- 短代码

object Solution 
    def twoSum(nums: Array[Int], target: Int): Array[Int] = 
        
        val map = nums.zipWithIndex
        map.filter(elem => map.exists(kv=>kv._1==target - elem._1&&kv._2!=elem._2)).map(_._2)

    

以上是关于LeetCode two sum -- scala 题解与思路的主要内容,如果未能解决你的问题,请参考以下文章

Leetcode Two Sum

[LeetCode] 1 Two Sum

[LeetCode] 1 Two Sum

LeetCode1. Two Sum 解题报告

leetcode 1.Two sum

Leetcode 1. Two Sum (Python)