LeetCode热题100 第一题 两数之和 双指针算法实现(c/c++)
1.题目描述:
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
示例 1:
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
示例 2:
输入:nums = [3,2,4], target = 6
输出:[1,2]
示例 3:
输入:nums = [3,3], target = 6
输出:[0,1]
数据范围:
2 <= nums.length <= 1e4
-1e9 <= nums[i] <= 1e9
-1e9 <= target <= 1e9
只会存在一个有效答案
进阶:
你可以想出一个时间复杂度小于 O(n^2) 的算法吗?
原题链接
2.算法思路:
本题在力扣中,官方题解给出暴力循环枚举和哈希表两种方法,非官方题解绝大部分使用暴力枚举、哈希表或map等方法。而面对一组数字序列时,如果能将其转化成有序序列,往往双指针算法会是一个更简单、性能更好的选择。
我们可以观察到,题目给出的数据是一个无序的数组,同时,最后我们需要返回数组元素初始下标。
所以,我们创建tmp数组,并将nums数组的值赋给tmp数组。
vector<int> ans;//答案数组
vector<int> tmp=nums;
这一步的目的是使用tmp数组,记录数组元素的初始下标。
同时,为了应用双指针算法,我们使用sort函数将原数组nums按照升序排序。
sort(nums.begin(),nums.end());
此时,我们创建m,n两个指针,分别指向nums数组的头和尾(nums数组最大值和最小值的下标)。
int m=0,n=nums.size()-1;
重点 :
此时分为三种情况。
1.当nums[m]+nums[n]的值大于target时:
n- -,将n的位置往前移一位。
2.当nums[m]+nums[n]的值小于target时:
m++,将m的位置往后移一位。
3.当nums[m]+nums[n]的值等于target时:
此时nums[m]和nums[n]就是我们要找到的值,break。
while(m<n){
if(nums[m]+nums[n]==target){
break;
}else if(nums[m]+nums[n]<target){
m++;
}else if(nums[m]+nums[n]>target){
n--;
}
}
至此,我们已经得到了和为目标值target的两个整数:num[m]和nums[n]。
现在我们需要做的是,在记录原始位置的数组tmp中,找到nums[m]和nums[n]的下标。
遍历tmp数组,当tmp[i]=nums[m]时,即为找到nums[m]对应的下标i, 将i放进ans数组。
for(int i=0;i<tmp.size();i++){
if(tmp[i]==nums[m]){
ans.push_back(i);
break;
}
}
再次遍历tmp数组,当tmp[i]=nums[n]时,即为找到nums[n]对应的下标。
但此时,如果nums[m]和nums[n]相等,那么就会导致两次遍历tmp数组找到的下标是同一个下标。
例如
target: 6
tmp:[7,3,1,2,6,3]
nums:[1,2,3,3,6,7]
显然,nums[m]和nums[n]的值都是3,此时,两次遍历tmp得到的结果都是1。
为了避免这种情况,在第二次遍历时,可以从后往前遍历tmp数组。
for(int i=tmp.size()-1;i>=0;i--){
if(tmp[i]==nums[n]){
ans.push_back(i);
break;
}
}
最后,返回ans
return ans;
3.时间复杂度分析
排序+三次遍历:nlogn+3n
时间复杂度:O(n*logn)
4.完整代码
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> ans;
vector<int>tmp=nums;
sort(nums.begin(),nums.end());
int m=0,n=nums.size()-1;
while(m<n){
if(nums[m]+nums[n]==target){
break;
}else if(nums[m]+nums[n]<target){
m++;
}else if(nums[m]+nums[n]>target){
n--;
}
}
for(int i=0;i<tmp.size();i++){
if(tmp[i]==nums[m]){
ans.push_back(i);
break;
}
}
for(int i=tmp.size()-1;i>=0;i--){
if(tmp[i]==nums[n]){
ans.push_back(i);
break;
}
}
return ans;
}
};