Leetcode 1. 两数之和
链接:https://leetcode-cn.com/problems/two-sum/
题目描述:给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 的那 两个 整数,并返回它们的数组下标。你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。你可以按任意顺序返回答案。
基本思路:首先对数组进行排序,且在排序之前获得接下来排序过后相对应的索引值(因为排序会导致原数组索引值改变)。然后对其中一个数使用遍历数组的方式寻找,对另一个数使用二分查找的方式寻找得到,使算法复杂度为O(nlogn)。
在获得sort数组排序相对应索引值的时候,使用了C++ 11 Lambda表达式。关于Lambda的使用可以参看下面两篇博文:
1、一个C++ Lambda实现的返回排序索引的模板函数
c++ ——排序并保留索引
2、Lambda基本知识的讲解
C++ 11 Lambda表达式
代码如下:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> result;
//对vector数组进行排序(方便后面使用二分法),且获得排序之后相对应索引值
vector<int> sortIdx(nums.size());
iota(sortIdx.begin(), sortIdx.end(), 0);//全部赋值为0
sort(sortIdx.begin(), sortIdx.end(), [&nums](int i1, int i2) {return nums[i1] < nums[i2];});//获得排序索引
sort(nums.begin(), nums.end());
//对其中一个值通过遍历整个数组寻找
for (int idx1 = 0; idx1 < nums.size() - 1; ++idx1) {
int another = target - nums[idx1];
result.push_back(sortIdx[idx1]);
//对另一个值通过二分查找
int start = idx1 + 1;
int end = nums.size() - 1;
while (start + 1 < end) {
int mid = start + (end - start) / 2;
if (another == nums[mid]) {
end = mid;
} else if (another > nums[mid]) {
start = mid;
} else if (another < nums[mid]) {
end = mid;
}
}
if (another == nums[start]) {
result.push_back(sortIdx[start]);
return result;
}
if (another == nums[end]) {
result.push_back(sortIdx[end]);
return result;
}
result.clear();
}
return result;
}
};