专栏持续更新50道算法题,都是大厂高频算法题,建议关注, 一起巧‘背’算法!
题目
解法一 暴力法(双层for循环)
如果没有刷过算法的,基本都会立马想到双层for循环。只要在内层循环里挨个跟外层相加,等于目标值,返回目标值的target即可。这种方式我又称为暴力解法,原因是基本是穷举完了,两层循环时间复杂度O(N2)。想不到其他方法时,可以跟面试官说暂时就想到了暴力解法,不要说双层for循环。原因是暴力法是一种穷举的算法思想,至少体现了你之前学过算法
public int[] twoSum(int[] nums, int target) {
int[] res = new int[2];
// 边界判断,数组中小于两个数字直接返回
if (nums.length < 2){
return res;
}
// 双层for循环
for(int i = 0; i< nums.length; i++){
for (int j = 0; j<nums.length;j++){
if (i == j){
continue;
}
if (target == nums[i] + nums[j]) {
res[0] = i;
res[1] = j;
}
}
}
return res;
}
解法二 数组转Map
实际开发中为了避免多层循环,通常会将List转Map。这里思想差不多,利用Map的特性可以将时间复杂度变为O(N)。思路是以元素为Key,以数组下标为value,遍历数组时,判断当前map内是否包含target减去当前遍历元素,如果存在则直接返回两个下标。不存在则把当前遍历值放到map中
public int[] twoSum2(int[] nums, int target) {
int[] res = new int[2];
// 边界判断,数组中小于两个数字直接返回
if (nums.length < 2){
return res;
}
// 利用map特性降低时间复杂度
Map<Integer,Integer> map = new HashMap<>();
for (int i = 0;i<nums.length;i++){
Integer num = nums[i];
// 判断map中是否存在值与当前值的和 等于target有则返回,没有将该数字放到map中
if (map.containsKey(target - num)){
res[0] = i;
res[1] = map.get(target-num);
}
map.put(num,i);
}
return res;
}
总结
第一次做这道题的时候,直接就想到了双层for循环,这也是大部分人立马能想到的解法。如果你看完没有一点思路那么就直接看题解,然后不看题解写下来慢慢去理解。解法二属于比较巧妙的解法,同样需要多写多看,多些几次,看看那里跟答案有差距,刷几次后慢慢就会有算法思维了