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 }

 

posted @ 2016-02-26 13:00  可普CS之家  阅读(174)  评论(0编辑  收藏  举报