题目描述

给定一个整数数组 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;

    }

}

巴特西

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

题目描述

解题思路

解题代码

最新文章

热门文章