这个学期开学就要开始找实习了呢......开始刷LeetCode准备一下面试吧
题目连接:
https://leetcode.com/problems/two-sum/
题目大意:
就是给出一个数组和一个target, 问数组中能够求和得到target的两个数的下标, 下标从0开始
大致思路:
首先有个很明确的写法是用vis[x]表示x是否出现,初始化-1, 然后从左往右对于当前的数x判断 target - x的vis数组是否不是-1, 不是-1就找到了,否则就标记vis[x] = 当前下标
不过上面这个做法由于数据范围的未知性不可写(感觉题目不说数据范围有点坑啊....),那么一个容易想到的替代方案便是用STL自带的map容器,map[x]表示x是否出现,不过时间复杂度提高到了O(nlogn),n是数组大小
注意两个很大的int数加起来可能超过了int造成伪解....比如给的target = -2^31, 数组是[2, 2^31 - 1, 0, -2^31])前面两个相加造成溢出
代码如下:
Result : Accepted Time : 24ms
class Solution { public: vector<int> twoSum(vector<int>& nums, int target) { map<long long, int> *M = new map<long long, int>(); for(int i = 0, sz = nums.size(); i < sz; i++){ long long need = (long long)target - (long long)nums[i]; if(M->count(need)){ delete M; return vector<int>{(*M)[need], i}; }else{ (*M)[nums[i]] = i; } } delete M;//记住防止内存泄露 return vector<int>{-1, -1};//Error, no answer found } };