leetCode–两数之和
题目描述:
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
1. 暴力法
解题思路
- 直接将数组循环两遍
- 判断两次循环取出的数字之和是否等于目标值
- 如果是则返回下标,不是返回null
代码示例
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;
}
}
测试结果:
2.Map集合法
解题思路
- 利用Map集合可以降低时间复杂度
- 首先创建一个空的Map集合
- 然后遍历整数数组取出元素
- 利用两数和减去取出的元素获得另外一个元素的值
- 如果这个值在集合中存在,利用键取出该值的下标一起返回
- 如果不存在就将这个值存到集合中继续重复取出数组元素
- 直到条件符合
解题代码
- 源代码
class Solution {
public int[] twoSum(int[] nums, int target) {
if(nums != null){
Map<Integer, Integer> map = new HashMap<>();
for(int i = 0;i < nums.length;i++){
int another = target - nums[i];
if(map.containsKey(another)){
return new int[]{map.get(another),i};
}else{
map.put(nums[i],i);
}
}
}
return null;
}
}
- 优化一下:
class Solution {
public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> map = new HashMap<>();
for(int i = 0;i < nums.length;i++){
if(map.containsKey(nums[i])){
return new int[]{map.get(nums[i]),i};
}
map.put(target - nums[i],i);
}
return null;
}
}
测试结果: