LeetCode －Two Sum

Posted 2020-06-16

　　题目要求很简单，给你一个数组（例如，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 }