思路:
1. 先对数组排序(升序),并记录原来数组中元素位置;
2. 分别设置头尾两个方向的迭代器(指针也可);
3. 如果头尾迭代器指示的两个元素相加之和小于目标值,则头迭代器(指针)后移;如果头尾两个方向的元素相加之和大于目标值,则头迭代器(指针)前移;
4. 循环步骤3,直到头尾迭代器相等;
5. 找到两个元素在原来数组中的位置,防止相同元素对应同一个位置。最后对这两个位置排个序,返回。
源代码如下:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> numsCopy = nums;
sort(numsCopy.begin(), numsCopy.end());
copy(numsCopy.begin(), numsCopy.end(), ostream_iterator<int>(cout, "\n"));
vector<int>::iterator iter = numsCopy.begin();
vector<int>::iterator iter_end = numsCopy.end();
iter_end--;
while (iter != iter_end && (*iter + *iter_end) != target) {
//cout << *iter << endl;
if (*iter + *iter_end < target) {
++iter;
}
else if (*iter + *iter_end > target) {
iter_end--;
}
else {
break;
}
}
vector<int> ret;
vector<int>::iterator firstPos;
int num1 = (find(nums.begin(), nums.end(), *iter)) - nums.begin();
ret.push_back(num1);
int num2 = find(nums.begin(), nums.end(), *iter_end) - nums.begin();
if (num1 == num2) {
num2 = find(nums.begin() + num1 + 1, nums.end(), *iter_end) - nums.begin();//防止在同一个位置,用了小技巧,因为只有一个结果,那么不会出现三个不同位置上的元素相等情况,最多是两个不同位置元素相等;
}
ret.push_back(num2);
sort(ret.begin(), ret.end());
return ret;
}
};