给定一个整数数组 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)