题目: 给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9 所以返回 [0, 1]
解法:
(1)O(n^2)算法
public class TwoSum {public int[] twoSum(int[] nums, int target) {for (int i = 0; i < nums.length; i++) {int value = target - nums[i];for (int j = i + 1; j < nums.length; j++) {if (nums[j] == value) {return new int[] {i, j};}}}return null;}}
(2)O(n)算法 Java版本
import java.util.HashMap;import java.util.Map;public class TwoSum {public int[] twoSum(int[] nums, int target) {Map<Integer, Integer> map = new HashMap<>();for (int i = 0; i < nums.length; i++) {int value = target - nums[i];if (map.containsKey(value)) {return new int[] {map.get(value), i};}map.put(nums[i], i);}return null;}}
O(n)算法Python版本
from typing import Listclass Solution:def twoSum(self, nums: List[int], target: int) -> List[int]:dict = {}for i, n in enumerate(nums):if target - n in dict:return [dict[target - n], i]dict[n] = i
