题目描述

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

你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。

示例:

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

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

思路

1.先将数组中的所有元素存放到哈希表中。
2.遍历数组中的全部元素,并检查哈希表中是否有答案,若有答案,则直接返回结果。

Java代码实现

    public int[] twoSum(int[] nums, int target) {
        Map<Integer,Integer> store = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            store.put(nums[i],i);
        }

        for (int i = 0; i < nums.length; i++) {
            if(store.containsKey(target-nums[i]) && store.get(target-nums[i]) != i)
                return new int[]{i,store.get(target-nums[i])};
        }

        return new int[]{-1,-1};
    }

Golang代码实现

func twoSum(nums []int, target int) []int {
    mymap := make(map[int]int)
    for i:=0; i<len(nums); i++ {
        if _,ok := mymap[target-nums[i]]; ok{
            return []int{i,mymap[target-nums[i]]}
        }else{
            mymap[nums[i]] = i
        }
    }
    return []int{-1,-1};
}