给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
这道题原理是 根据遍历当前值使用哈希查找目标值的下标 。在有“元素不能使用两遍”限制的情况下,可以通过创建Map[int][]int在加入相同元素时避免覆盖下标。在遍历数组的过程中出现相同值(Key)时,将其下标append进同一数组(Value)。
又因为输入只会对应一种答案,所以重复的值不会出现两个以上。这样结果成立的条件就是:Map中存在这个值且第一次出现的下标不等于当前下标 。
func twoSum(nums []int, target int) []int {
var p = make(map[int][]int)
for i, v := range nums {
p[v] = append(p[v], i)
if _, ok := p[target - v]; ok && p[target - v][0] != i {
return []int{p[target - v][0], i}
}
}
return nil
}
Java解法,只要在最后再添加元素进哈希表就不需要检查重复下标的问题
class Solution {
public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < nums.length; i++){
if (map.containsKey(target - nums[i]))
return new int[]{i, map.get(target - nums[i])};
map.put(nums[i], i);
}
throw new IllegalArgumentException("no two sum solution");
}
}