leetcode能用java刷吗_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 numMap = new HashMap();

//设置值和对应的index

for(int i=0;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

//判断两个数值相加是否为目标值

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 numMap = new HashMap();

for(int i=0;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;

}

}

代码仓库