LeetCode算法第一题–两数之和
问题描述
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
原因分析:
问题一:数组中同一元素在答案里不能重复出现
例如:
nums = [3,2,4]
target = 6
如果使用第一次的代码:
class Solution {
public int[] twoSum(int[] nums, int target) {
for(int i = 0;i<nums.length;i++){
for(int j=0;j<nums.length;j++){
if(nums[i]+nums[j]==target){
return new int[]{i, j};
}
}
}
return new int[0];
}
}
输出结果为[0,0]
预期输出应该是[1,2]
解决问题:
将第二个for循环进行修改
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 new int[0];
}
}
该方法时间复杂度为O(n^2),其中 n 是数组中的元素数量。最坏情况下数组中任意两个数都要被匹配一次。
改进算法:
使用哈希表。
降低时间复杂度为O(n),个人理解为将哈希表的索引当作第二个for循环。
class Solution {
public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> hashtable = new HashMap<Integer, Integer>();
for (int i = 0; i < nums.length; ++i) {
if (hashtable.containsKey(target - nums[i])) { //判断哈希表的键是否存在target-nums[i]。如果存在,返回通过Map的get方法获得对应键的值(即数组的索引)和当前数组的索引
return new int[]{hashtable.get(target - nums[i]), i};
}
hashtable.put(nums[i], i); //将数组的值作为键,索引作为值放入到哈希表中
}
return new int[0];
}
}
总结
第一次简单算法题过程艰难 2022-3-29