题目:
Given an array of integers, return indices of the two numbers such that they add up to a specific target.
You may assume that each input would have exactly one solution, and you may not use the same element twice.
- Example:
- Given nums = [2, 7, 11, 15], target = 9,
- Because nums[0] + nums[1] = 2 + 7 = 9,
- return [0, 1].
题意:
给定一个整数数组 nums 和一个目标值 target, 请你在该数组中找出和为目标值的那 两个 整数, 并返回他们的数组下标.
你可以假设每种输入只会对应一个答案. 但是, 你不能重复利用这个数组中同样的元素.
Solution:
方法一: 暴力法
Java:
- public int[] twoSum(int[] nums, int target) {
- for (int i = 0; i <nums.length; i++) {
- for (int j = i + 1; j < nums.length; j++) {
- if (nums[j] == target - nums[i]) {
- return new int[] { i, j };
- }
- }
- }
- throw new IllegalArgumentException("No two sum solution");
- }
复杂度分析:
• 时间复杂度: O(n^2), 对于每个元素, 我们试图通过遍历数组的其余部分来寻找它所对应的目标元素, 这将耗费 O(n) 的时间, 因此时间复杂度为 O(n^2).
• 空间复杂度: O(1).
方法二: 一遍哈希表
为了对运行时间复杂度进行优化, 我们需要一种更有效的方法来检查数组中是否存在目标元素. 如果存在, 我们需要找出它的索引.
保持数组中的每个元素与其索引相互对应的最好方法是什么? 哈希表.
通过以空间换取时间的方式, 我们可以将查找时间从 O(n) 降低到 O(1). 哈希表正是为此目的而构建的, 它支持以 近似 恒定的时间进行快速查找.
我用 "近似" 来描述, 是因为一旦出现冲突, 查找用时可能会退化到 O(n). 但只要你仔细地挑选哈希函数, 在哈希表中进行查找的用时应当被摊销为 O(1).
在进行迭代并将元素插入到表中的同时, 我们还会回过头来检查表中是否已经存在当前元素所对应的目标元素. 如果它存在, 那我们已经找到了对应解, 并立即将其返回.
C++:
- class Solution {
- public:
- vector<int> twoSum(vector<int>& nums, int target) {
- unordered_map<int, int> mp;
- int i = 0;
- for (; i <nums.size(); ++i) {
- if (mp[target - nums[i]] != 0) {
- //found
- vector<int> r = {mp[target - nums[i]] - 1, i}; // 由于 mp[nums[i]] 先前加了 1, 所以要减去 1
- return r;
- } else { //if(mp[target - nums[i]] == 0), 说明哈希表不存在这个值, 就要把这个值存入哈希表中.
- mp[nums[i]] = i + 1; // 由于 i = 0, 避免 mp[nums[i]] 也等于 0
- }
- }
- vector<int> r;
- return r;
- }
- };
Java:
- public int[] twoSum(int[] nums, int target) {
- Map<Integer, Integer> map = new HashMap<>();
- for (int i = 0; i < nums.length; i++) {
- int complement = target - nums[i];
- if (map.containsKey(complement)) {
- return new int[] { map.get(complement), i };
- }
- map.put(nums[i], i);
- }
- throw new IllegalArgumentException("No two sum solution");
- }
复杂度分析:
时间复杂度: O(n), 我们只遍历了包含有 n 个元素的列表一次. 在表中进行的每次查找只花费 O(1) 的时间.
空间复杂度: O(n), 所需的额外空间取决于哈希表中存储的元素数量, 该表最多需要存储 n 个元素.
来源: http://www.bubuko.com/infodetail-2985546.html