Leetcode001 两数之和

题目001 两数之和

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

public class Solution{
    public int[] twoSum(int[] nums,int target){
        int[] res=new int[2];
        if(nums==null || nums.length<=1)return res;
            HashMap<Integer,Integer> map=new HashMap<Integer,Integer>();
            for(int i=0;i<nums.length;i++){
                int num=nums[i];
		int val=target-num;
		if(map.containsKey(val)){
		    res[0]=i;
		    res[1]=map.get(val);
		    return res;
		}else{
		    map.put(num,i);
		}
	    }
	    return res;	
	}
}

Python代码

class Solution(object):
    def twoSum(self, nums, target):
        hashmap={}
        for i,num in enumerate(nums):
            if hashmap.get(target-num) is not None:
                return [hashmap.get(target-num),i]
            hashmap[num]=i
if __name__=="__main__":
    nums=[2,7,11,15]
    target=26
    s=Solution()
    print(s.twoSum(nums,target))