LeetCode--1.两数之和

1 题目描述

给定一个整数数组 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]

2 解题思路

2.1 暴力破解

两层 for 循环, 外层循环 0 ~ nums.length - 1, 内层循环 外层 + 1 ~ nums.length, 这样虽然能简单粗暴解题, 但时间复杂度 O($n^2$)

2.2 使用 Map

1. 目标是 nums[i] + nums[j] = target, 转换一下变成 target - nums[i] = nums[j]
2. 在遍历 nums 时, 将数据依次存储到 map, key为 nums[i], value 为位置 i
3. 使用 target - nums[i] 的结果去 map 中查找, 看是否能匹配上, 这样就能快速找到 nums[j]

时间复杂度为O(n)

class Solution {
	public int[] twoSum(int[] nums,  int target) {  
	    int[] result = new int[2];  
	    // 如果数组为空或者长度小于2, 直接返回  
	    if (nums == null || nums.length < 2) {  
	        return result;  
	    }  
	    // key为数组元素, value为数组下标,  保存已经遍历过的元素  
	    Map<Integer,  Integer> dataMap = new HashMap<>(nums.length);  
	      
	    for (int i = 0; i < nums.length; i++) {  
	        // 如果 map 中包含target - nums[i], 则说明找到了  
	        if (dataMap.containsKey(target - nums[i])) {  
	            result[0] = dataMap.get(target - nums[i]);  
	            result[1] = i;  
	            break;  
	        }  
	        // 如果 map 中不包含target - nums[i], 则将当前元素放入map中  
	        dataMap.put(nums[i],  i);  
	    }  
	    return result;  
	}
}