LeetCode(1) -Two Sum
题目要求很简单,给你一个数组(例如,nums = [2,7,11,15])和一个target(target = 9),找到数组里两个数相加后能得到target的这两个数的index。在本例中,返回的应该是[0,1]。
碰到这样的题目,首先应该想到的是运用HashMap,记录数组里的每个元素和对应的index,把O(n^2)的问题转变为O(n)的问题,然后再第二次遍历数组,当current_number = nums[i]时,寻找HashMap里面是否存在target - nums[i],若存在,则输出的index为[i, map.get(terget - nums[i])]。
代码如下:
1 public class Solution { 2 public int[] twoSum(int[] nums, int target) { 3 HashMap<Integer,Integer> map = new HashMap<Integer,Integer>(); 4 for (int i = 0; i < nums.length; i++) { 5 map.put(nums[i],i); 6 } 7 int[] res = new int[2]; 8 for (int i = 0; i < nums.length; i++) { 9 int remain = target - nums[i]; 10 if (map.containsKey(remain) && map.get(remain) != i) { 11 res[0] = i; 12 res[1] = map.get(remain); 13 break; 14 } 15 } 16 return res; 17 } 18 }