最新消息:深度思考

two sum

算法 liuxuecheng 2570浏览 0评论

1.题目

给定一个整型数组array和一个整数target,在数组中找出两个整数相加之和为target,返回这两个整数的下标。

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

2.解法

  1. 最容易想到的方法就是使用两层for循环去遍历,但是这种方法时间复杂度为O(n^2)
public int[] twoSum(int[] array, int target) {
        int[] result  = new int[2];
        for (int i = 0; i < array.length; i++) {
            for (int j = i + 1; j < array.length; j++) {
                if (array[i] + array[j] == target) {
                    result[0] = i;
                    result[1] = j;
                }
            }
        }
        return result;
  }
  1. 一种较为优化的解法是使用HashMap,key为当前整型数num,value为下标index,判断target-num是否在当前HashMap中,若在则返回num的下标同时从HashMap中取出target-num的下标,否则将num放入HashMap。
public int[] twoSum(int[] array,int target){
    HashMap<Integer,Integer> map = new HashMap<Integer,Integer>();
    for(int i = 0;i<array.length;i++){
        int tmp = target-array[i];
        if(map.containsKey(tmp)){
            return new int[]{i,map.get(tmp)};
        }else{
            map.put(array[i],i);
        }
    }
    throw new IllegalArgumentException("not find two num");
}

这个问题在后面做了引申,列出来一堆相似的问题,链接:http://bigdataer.net/?p=600

转载请注明:大数据随笔 » two sum

发表我的评论
取消评论

表情

Hi,您需要填写昵称和邮箱!

  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址