今年准备刷一下leetcode,提升一下算法相关的能力。从最简单的开始刷起,各位大佬不要嘲笑我。不知道能坚持多久。
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
你可以按任意顺序返回答案
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
输入:nums = [3,2,4], target = 6 输出:[1,2]
输入:nums = [3,3], target = 6 输出:[0,1]
1、先用一个list存储数组
2、由于只有一个解,所以只需要一次遍历即可,为了防止有重复元素的出现,所以再找第二个下标时先把第一个下标对应的值设置成Integer.MAX_VALUE
3、如果找不到第二个下标,将第一个下标对应的值复原。
public int[] twoSum(int[] nums, int target) {
List<Integer> num_list = new ArrayList<Integer>();
for (int num : nums) {
num_list.add(num);
}
for (int i=0;i<nums.length;i++) {
int curr = num_list.get(i);
num_list.set(i,Integer.MAX_VALUE);
if (num_list.contains(target - curr)) {
return new int[]{i,num_list.indexOf(target - curr)};
}
num_list.set(i,curr);
}
return new int[2];
}
public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> map = new HashMap<>();
for(int i = 0; i< nums.length; i++) {
if(map.containsKey(target - nums[i])) {
return new int[] {map.get(target-nums[i]),i};
}
map.put(nums[i], i);
}
throw new IllegalArgumentException("No two sum solution");
}