从零开始刷算法----两数之和

89 阅读1分钟

「这是我参与2022首次更文挑战的第1天,活动详情查看:2022首次更文挑战

题目描述

给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。 你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。 你可以按任意顺序返回答案。 示例 1: 输入:nums = [2,7,11,15], target = 9 输出:[0,1] 解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。 示例 2: 输入:nums = [3,2,4], target = 6 输出:[1,2] 示例 3: 输入:nums = [3,3], target = 6 输出:[0,1]

思路介绍

思路一

使用双重for循环,循环遍历数组,获得nums[i],nums[j]两个值,

首先需要判断是否存在nums[i],nums[j]相加等于target,如果存在,直接返回两个下标值,结束循环;

如果不存在nums[i],nums[j]相加等于target,进入下一个循环,直至到找到两个数相加等于目标值;

或者遍历完整个数组,没有找到两个数相加等于目标值的结果。

代码

  public static int[] twoSum1(int[] nums, int target) {
        int n = nums.length;
        for (int i = 0; i < n; ++i) {
            for (int j = i + 1; j < n; ++j) {
                if (nums[i] + nums[j] == target) {
                    return new int[]{i, j};
                }
            }
        }
        return null;
    }

思路二

使用哈希map进行解题。

map是一种key value的存储结构,可以用key保存数值,用value在保存数值所在的下标。

使用双重for循环也可以很快解决,时间复杂度O(N2)。

而map中提供containsKey 方法,可以一次遍历解决问题,时间复杂度较低O(N),所以选择使用哈希容器map处理

遍历数组nums ,i为当前下班,数组每个元素都判断map中是否存在target-nums[i]的key值

如果存在则找到了两个值,如果不存在则将当前的 (nums[i],i) 存入 map 中,继续遍历直到找到为止

如果最终都没有结果则返回null

image.png

代码

 public static int[] twoSum(int[] nums, int target) {
        Map<Integer,Integer> map = new HashMap<>();
        int[] newNums = new int[2];
        for(int i = 0 ;i<nums.length;i++){
            int cnum = target-nums[i];
            if(map.containsKey(cnum)){
                newNums[0] =i;
                newNums[1] = map.get(cnum);
                return newNums;
            }else {
                map.put(nums[i],i);
            }
​
        }
        return null;
    }

总结

算法能力还是比较差,最先想到的是暴力拆解,看到其他人的解法才想起来还可以使用map,然后自己分析了下map的解题过程