运行结果
一、题目
二、分析
时间复杂度:O(n)
空间复杂度:O(n)
题目没有对空间做要求,所以我们应该采用以空间换时间的思路。
题目要求找到符合要求的元素的下标。我们在结题时无论采用哪种方法,一定是先找到符合题目要求的两个元素。然后将其对应的下标作为返回值。这种有对应关系的数据与键(key)值(value)对很类似。使用哈希表会非常容易。
步骤:
- 首先对空数组以及数组长度为1 的情况进行排除。
- 我们需要使用一个Map集合,其中键(key)表示数组中的元素,值(value)表示该元素在数组中的下标。之所以这样表示是因为我们的返回值是元素的下标。利用 Map 的 get ( ) 方法可以获得。
- 对数组进行循环。每次循环会得到一个元素,即 nums [i] ,接下来判断Map集合中是否存在 key = target - nums [i] 的键。使用的方法为 containsKey ( ) 。①如果存在,直接返回 i 和 该key对应的value值。②如果不存在,在Map中添加该元素与其对应的下标。进行下一次比较。
- for循环结束之后记得 return null ; 表示数组中没有符合要求的元素。如果不写编译不通过。
三、源码
class Solution {
public int[] twoSum(int[] nums, int target) {
if(2 > nums.length){
return null;
}
Map<Integer , Integer> m = new HashMap<>();
for(int i = 0; i < nums.length; i++){
int x = target - nums[i];
if(m.containsKey(x)){
return new int[]{m.get(x) , i};
}else{
m.put(nums[i] , i);
}
}
return null;
}
}