LeetCode1. 两数之和(映射的应用)

1. 两数之和

LeetCode第1号问题,题目链接:https://leetcode-cn.com/problems/two-sum/)

给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

题解:最先想到的就是使用双重循环暴力解法,第一层循环遍历标记i的位置,第二层循环标记j的位置,需要注意的是j需要从i的后一个位置开始,因为说明了前提数组中同一个元素不能使用两遍,所以定义j=i+1,当遍历找到两数之和等于目标值target时,返回他们的下标。

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

但是上述解法存在一定的缺陷即是使用了双重循环,时间复杂度为O(N^2)。可以在遍历第一个数的时候就根据target来求得它配对的数target-nums1。如果target是nums1的2倍,则需要num1的值在数组中出现了两次。

  1. 创建一个映射Map,用于存放k为值,v为下标的键值对
  2. 遍历数组,对每一个元素num1都进行target-num1求它的配对值num2
  3. 如果不存在,存入num1和它的下标。
  4. 如果配对值num2存在映射的key中,则返回当前遍历的下标和num2的value
class Solution {
    public int[] twoSum(int[] nums, int target) {
        HashMap<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[]{map.get(num), i};
            map.put(nums[i], i);
        }
        return null;
    }
}

猜你喜欢

转载自blog.csdn.net/wankcn/article/details/107770328