leetcode刷题笔记-1. 两数之和（java实现）

题目描述

给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那 两个 整数，并返回他们的数组下标。

你可以假设每种输入只会对应一个答案。但是，数组中同一个元素不能使用两遍。

示例:

给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

来源：力扣（LeetCode） 链接：https://leetcode-cn.com/problems/two-sum

解题思路

把数组中的值，存储到hashmap中，key为值，value放置index，当遍历时发现组队相加为目标值时，退出循环，返回数值对应的index。

解题代码

下边的代码，执行用时：8 ms  内存消耗：39 MB。经过思考后发现还可以再优化。

class Solution {
    public int[] twoSum(int[] nums, int target) {
        Map<Integer, String> numMap = new HashMap<Integer, String>();
        //设置值和对应的index
        for(int i=0;i<nums.length;i++) {
            if(numMap.containsKey(nums[i])) {
                numMap.put(nums[i], numMap.get(nums[i])+"_"+i);
            } else {
                numMap.put(nums[i], i+"");
            }
        }
        int[] result = new int[2];
        for(int i=0;i<nums.length;i++) {
            //判断两个数值相加是否为目标值
            if(numMap.containsKey(target-nums[i])) {
                if((target-nums[i]) == nums[i] && !numMap.get(nums[i]).contains("_")) {
                    //两个数值是相等的，但是数组中该值只有一个
                    continue;
                } else if((target-nums[i]) == nums[i] && numMap.get(nums[i]).contains("_")) {
                    //满足要求，记录index，返回结果结束
                    String[] is = numMap.get(target-nums[i]).split("_");
                    result[0] = Integer.parseInt(is[0]);
                    result[1] = Integer.parseInt(is[1]);
                    break;
                }
                result[0] = i;
                result[1] = Integer.parseInt(numMap.get(target-nums[i]));
                break;
            }
        }
        return result;
    }
}

优化后的代码，执行用时：3 ms  内存消耗：37.8 MB。

class Solution {
    public int[] twoSum(int[] nums, int target) {
        int[] result = new int[2];
        //存储数值和下标的hashmap
        Map<Integer, Integer> numMap = new HashMap<Integer, Integer>();
        for(int i=0;i<nums.length;i++) {
            //判断与当前值相加等于目标数值的数是否在hashmap中
            //如果存在则取当前下标，和hashmap中的下标，返回
            //如果不存在把数值作为key，index作为value，存储到hashmap中
            if(numMap.containsKey(target-nums[i])) {
                result[1] = i;
                result[0] = numMap.get(target-nums[i]);
                break;
            } else {
                numMap.put(nums[i], i);
            }
        }
        return result;
    }
}