前言:
元旦放假实在无聊,写点代码娱乐自己吧,平时工作负责项目管理方面的东西相对多了一些,作为一个热爱编程的人实在是手痒,趁着假期练练手,当然了,不是为了刷题,就是为了让自己保持一个编程的状态。个人觉得为了刷题而刷题会失去很多写代码的乐趣,由于自己水平有限,如果代码有问题或者有什么更好的办法也欢迎各位大神提意见,感谢。
题目来源:https://leetcode-cn.com/problems/two-sum/submissions/
正文:
给定一个整数数组 nums
和一个目标值 target
,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9 因为 nums[0] + nums[1] = 2 + 7 = 9 所以返回 [0, 1]
1:第一次随意写写,使用了最简单的暴力穷举法,代码如下:
class Solution {
public int[] twoSum(int[] nums, int target) {
int[] targetNum = new int[2];
for (int i = 0; i < nums.length; i++) {
for (int k = i+1; k < nums.length ; k++) {
if (nums[i] + nums[k] == target) {
targetNum[0] = i;
targetNum[1]=k;
}
}
}
return targetNum;
}
}
该算法的时间复杂度为O(n2),结果如下:
2.由于上面的算法实在是太慢,自己优化了一下,大概想法是使用hash匹配,代码如下:
class Solution {
public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> hashMap = new HashMap();
//int record = 0;
for (int i = 0; i < nums.length; i++) {
hashMap.put(nums[i], i);
}
for (int j = 0; j< nums.length; j++) {
int record = target - nums[j];
Integer result = hashMap.get(record);
if (result != null && result != j) {
return new int[] { j, result };
}
}
return null;
}
}
该算法的时间复杂度为O(n),结果如下: