题目来源于leetcode,每日一刷小算法。

不管遇到什么挫折,明天的太阳都会照样升起

题目

给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。

你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。

示例

给定:nums = [2, 7, 11, 15], target = 9

因为:nums[0] + nums[1] = 2 + 7 = 9

所以返回 [0, 1]

思路

暴力解法

先用一个for循环把数组遍历出来,里面再套一个循环去寻找下一个数。

当外层循环在0位置时,内层循环从1位置开始搜索,如果外层+内层的数刚好等于target,就返回这两个位置(数组下标)。

示例:外层循环的在0的位置,内层循环在1的位置,此时他们两个是:2+7 == 9

所以将0和1返回。

但是这种解法效率很一般,不推荐使用。

Map解法

依旧使用for循环把数组遍历一遍,target-每个数 的结果去map里面搜索,如果没有的话,把这个数保存起来。如果有的话,此时正好返回for循环中的i下标和map中的值下标。

所以这个map我们应该设计成

key:数值

value:下标

示例:for循环开始后,9-2=7,此时7不在map中,将2保存起来。第i次(第二次)循环,9-7=2,此时2正好在map中。返回2对应的value和for循环中的i

从结果看,这种解法更加高效!

代码

暴力解法

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;
    }
}

Map解法

class Solution {
    public int[] twoSum(int[] nums, int target) {
        Map<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[]{i,map.get(num)};
            }
            map.put(nums[i],i);
        }
        return null;
    }
}

Last modification:February 18, 2021
如果觉得我的文章对你有用,请随意赞赏