题意:给一个int数组和一个int目标数,返回一个带两个数的数组。这两个数是输入数组中两个数的标签,标签指向的两个数相加得到目标数。
思路:常见的有O(n²)和O(n)两种解法,前者即暴力搜索,按顺序先找一个数挨个对着加起来找正确的和。
这里只说后者0(n)解法。此时需用哈希表存储数组和索引来实现。
for语句中,对于第i个数字,哈希表查找是否存在目标数减去第i个数字的差值这个数。如果找到,则就已经得出答案,一个索引是i,一个索引是差值的索引。
如果没找到,说明比这个数小的数(即已经记录到map中的数)没有等于差值的,有可能,这个差值在后面会有,当i排到这个差值的时候,原来这个第i个数就是当前第i个数的差值了。
没找到的情况下,就把第i个数和他的标签存到哈希表中,然后继续循环。
import java.util.HashMap;
import java.util.Map;
public class Solution {
public int[] twoSum(int[] nums, int target) {
int[] res = new int[2];
Map<Integer, Integer> map = new HashMap<Integer, Integer>();
for(int i = 0; i<nums.length; i++)
{
if(map.containsKey(target - nums[i]))
{
res[1] = i;
res[0] = map.get(target - nums[i]);
return res;
}
map.put(nums[i], i);
}
return res;
}
}