1. 两数之和
LeetCode第1号问题,题目链接:https://leetcode-cn.com/problems/two-sum/)
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
题解:最先想到的就是使用双重循环暴力解法,第一层循环遍历标记i的位置
,第二层循环标记j的位置,需要注意的是j需要从i的后一个位置开始
,因为说明了前提数组中同一个元素不能使用两遍,所以定义j=i+1
,当遍历找到两数之和等于目标值target时,返回他们的下标。
class Solution {
public int[] twoSum(int[] nums, int target) {
for (int i = 0; i < nums.length; i++) {
for (int j = i + 1; j < nums.length; j++) {
if (nums[i] + nums[j] == target)
return new int[]{i, j};
}
}
return null;
}
}
但是上述解法存在一定的缺陷即是使用了双重循环,时间复杂度为O(N^2)。可以在遍历第一个数的时候就根据target来求得它配对的数target-nums1
。如果target是nums1的2倍,则需要num1的值在数组中出现了两次。
- 创建一个映射Map,用于存放k为值,v为下标的键值对
- 遍历数组,对每一个元素num1都进行
target-num1
求它的配对值num2 - 如果不存在,存入num1和它的下标。
- 如果配对值num2存在映射的key中,则
返回当前遍历的下标和num2的value
。
class Solution {
public int[] twoSum(int[] nums, int target) {
HashMap<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < nums.length; i++) {
int num = target - nums[i];
if (map.containsKey(num))
return new int[]{map.get(num), i};
map.put(nums[i], i);
}
return null;
}
}