【LeetCode】1. Two Sum(Java)

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/Rabbit_Judy/article/details/87915116

题目描述：给定一个整数数组，返回两个数字的索引，使它们相加成为一个特定的目标。

您可能假设每个输入只有一个解决方案，并且您可能不会两次使用相同的元素。

Example:
Given nums = [2, 7, 11, 15], target = 9,
Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].

import java.util.Arrays;
import java.util.HashMap;

/**
 * 两数之和
 * @author wangfei
 *
 */
public class Solution {

	/**
	 * 将数组中的 元素存储在map中，并查看target减去数组中的一个元素之后其差值是否也存在于map中,
	 * 若存在，返回这两个数组元素的下标以数组的形式。
	 * @param nums
	 * @param target
	 * @return
	 */
	public static int[] twoSum(int[] nums, int target) {
        int[] res = new int[2];
        if(nums == null || nums.length<2)
            return res;
        HashMap<Integer,Integer> map = new HashMap<Integer, Integer>();
        for(int i=0; i<nums.length; i++) {                      
            if(map.containsKey(target-nums[i])) {
                res[0] = map.get(target-nums[i]);
                res[1] = i;
                return res;
            }
            map.put(nums[i],i); 
        }
        return res;
    }
	
	public static void main(String[] args) {
		int[] nums = {2, 7, 11, 15};
		int target = 9;
		System.out.println(Arrays.toString(twoSum(nums, target)));
	}
}