給定一個整數數組和一個目標值,找出數組中和為目標值的兩個數。
你可以假設每個輸入只對應一種答案,且同樣的元素不能被重復利用。
示例:
給定 nums = [2, 7, 11, 15], target = 9
因為 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
分析:
題意容易理解,找出數組中兩個元素,他們之和等于傳入的target參數,最后返回這兩個元素的腳標即可。
解法一:
可以使用最樸素的方法,用類似選擇排序的方法進行遍歷,對所有的元素進行相加操作,并且判斷結果是否等于target。但這種方法時間復雜度高,速度慢。
Java解答如下:
public int[] twoSum(int[] nums, int target) {
int[] result = new int[2];
for (int i = 0; i < nums.length; i++) {
for (int j = i + 1; j < nums.length; j++) {
if (nums[i] + nums[j] == target) {
result[0] = i;
result[1] = j;
return result;
}
}
}
return result;
解法二:
可以借助HashMap來減小運算的時間復雜度。遍歷過的元素和對應的腳標都存放到HashMap中,新遍歷到的元素,判斷HashMap中有沒有可以和它相加等于target的值。如果HashMap中包含有,就得到了結果。
Java解法如下:
public int[] twoSum(int[] nums, int target) {
HashMap<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < nums.length; i++) {
if (!map.isEmpty()) {
Integer index = map.get(target - nums[i]);
if (index != null) {
return new int[]{index, i};
}
}
map.put(nums[i], i);
}
return null;
}