在一上午摸索无果之后果断选择来看一手题解,结果因为题解中的unordered_map之前没接触过,先去看了网上的一些文章.首先罗列一下 map,unordered_map,set,unordered_set的区别和联系:
map
底层是红黑树,插入删除,查询操作时间复杂度都是O(logn)
unordered_map
底层是hash表,查询的时间复杂度为O(1),效率比map高,但是空间复杂度也比map高.
set,unordered_set
类似于上面的map和unordered_map区别,注意这里在set中插入同样的元素是无效的.!
回到这一题中,输入比如3,3 target=6时,最后返回的是0,1
所以我们需要使用能够插入+查询同样元素的高效数据结构,所以我们选择unordered_map.
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int, int> map; //使用散列表,底层直接取余
for (int i = 0; i < nums.size(); ++i) {
auto iter = map.find(target - nums[i]); //unordered_map.find(..) 就是寻找key-value中的key值!
if (iter != map.end()) {
return { iter->second,i };
}
map[nums[i]] = i;
}
return {};
}
我们注意以下几点
1)最后如果没有满足条件的,就直接返回'{}'
2)我们每次向map中插入元素是通过map[nums[i]]=i 建立映射关系,因为底层使用散列表,所以不用担心下标越界(比vector就不能做到散列表这样的任意下标访问)
3)我们每一次插入数据之前都要判断当前map中是否有能够凑成对的元素,如果有就直接返回,而不是先全部插入,不然会有重复的元素插入unordered_map,这会导致错误