1.题目描述
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2. 解法
解法一:
思想: 在数组中寻找和为目标值的数,返回其下表。可以选用暴力解法,直接两次遍历数组,寻找和为目标值的数,返回其下表。
代码:
public static int[] twoSum(int[] nums, int target) {
for (int i = 0; i < nums.length -1; i++) {
for (int j = i+1; j < nums.length; j++) {
if (nums[i] + nums[j] == target) {
int[] res = new int[]{i,j};
return res;
}
}
}
return new int[0];
};
结果:
解法二:
分析:
解法1虽然能解决问题,但是时间复杂度为 O(N^2)。考虑一种更好的解法。思考,我想要的是数组中和为 target 的两个数,通过一次遍历,我能依次拿到其中的每个数num[i],那么只需要查找剩下的数据中是否存在值为 target-num[i] 的数。 已知利用HashMap 哈希映射的特点,可以直接查询数据中是否包含某个 key,所以将数组中的数据以<num[i] ,i>的形式插入到 hashMap 中,实现一次遍历数组解决问题。具体如下:
- 遍历数组,获取下标位 i 的数,
- 查询HashMap 中是否 含有 key 为 target-num[i] 的数据,如果存在 获取其 value ,并返回 i 和value,如果不存在,则向HashMap 中以 <num[i] ,i>的形式插入数据
代码
public static int[] twoSum(int[] nums, int target) {
Map<Integer,Integer> numsHashMap = new HashMap<Integer, Integer>();
for (int i = 0; i < nums.length; i++) {
if(numsHashMap.containsKey(target-nums[i])){
return new int[]{i,numsHashMap.get(target-nums[i])};
}
numsHashMap.put(nums[i],i);
}
return new int[0];
};