LeetCode 1. 两数之和

303 阅读1分钟

leetcode.cn/problems/tw…

给定一个整数数组 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) 的算法吗?

解法一:

暴力求解,复杂度为 O(N²)。

class Solution {
    fun twoSum(nums: IntArray, target: Int): IntArray {
        val result = mutableListOf<Int>()

        for (i in nums.indices) {
            val x = nums[i]
            for (j in i + 1 until nums.size) {
                val y = nums[j]
                if (x + y == target) {
                    result.add(i)
                    result.add(j)
                }
            }
        }
        return result.toIntArray()
    }
}

解法二:

利用 Map 保存下标与下标对应的值,复杂度 O(N)。

class Solution {
    fun twoSum(nums: IntArray, target: Int): IntArray {
        val map = mutableMapOf<Int, Int>()

        for ((index, i) in nums.withIndex()) {
            map[i] = index
        }

        for ((index, i) in nums.withIndex()) {
            val y = target - i
            if (map.containsKey(y)) {
                val yIndex = map[y]
                yIndex?.let {
                    if (index < it) {
                        val result = intArrayOf(index, yIndex)
                        return result
                    }
                }
            }
        }
        return intArrayOf()
    }
}
  • 时间复杂度: O(n)
  • 空间复杂度: O(n)