从今天开始刷LeetCode,今天刷的第一题是:
题目的关键是两数相加等于target ,只有一个solution ,所以返回的数组一定只有2个下标,以下为我的代码:
public class Solution {
public int[] twoSum(int[] nums, int target) {
//创建一个数组保存符合题目要求的index值
int[] res= new int[2];
//已知数组从取出下标为1的和第二个、第三个、第四个...数字依次相加与target数字相比是否相等,相等即立刻记录下标值,退出循环
for(int i = 0;i<nums.length;i++) {
for(int j = 0;j<nums.length;j++) {
//两个相加的值与target相比是否相等,并且两个数的下标不能相等,保证了不会出现同一个下标的两个数值相加
if (nums[i] + nums[j] == target && i != j) {
res[0]=i;
res[1]=j;
break;
}
}
}
return res;
}
虽然以上代码比较容易明白,但是从效率上很有问题,因此还有另外一种方式,利用map集合,只要遍历数组一次即可以得出答案
public int[] twoSum(int[] numbers, int target) {
int[] result = new int[2];
Map<Integer, Integer> map = new HashMap<Integer, Integer>();
for (int i = 0; i < numbers.length; i++) {
if (map.containsKey(target - numbers[i])) {
result[1] = i + 1;
result[0] = map.get(target - numbers[i]);
return result;
}
map.put(numbers[i], i + 1);
}
return result;
}