解法1:先排序后用双指针。这里用逆向双指针好像不太可行,因为假设target=6, 两个符合条件的数的索引可能是(1,7), (3,9), (7,12),等等。假如我们设p1=0, p2=12,但是nums[p2]-nums[p1]>6,那我们是p1++还是p2–呢? 两个都进行就复杂了。
这里用同向双指针比较好。一开始p1=0, p2=1。当两指针指向的值的difference
这里要先用一个class把数组的index保存下来,因为sort后顺序会乱。另外一个要注意的地方是如果用stl::sort的话,必须把less_pair()定义成static,因为sort是global function(),似乎只能调用global或static function?
代码如下:
#include
#include
#include
using namespace std;
class PairClass {
public:
PairClass(int i=0, int v=0) : index(i), val(v) {}
int index;
int val;
};
static bool less_pair(const PairClass &first, const PairClass &second) {
return first.val < second.val;
}
vector twoSum7(vector &nums, int target) {
int len=nums.size();
vector result;
if (len<=1) return result;
int p1=0, p2=1;
//initialize pairs
vector pairs(len);
for (int i=0; i nums={7,2,15,24};
//vector ret=twoSum7(nums, -5);
//for (int i:ret) cout< nums={1,0,1};
//vector ret=twoSum7(nums, 0);
//for (int i:ret) cout< nums={1,2,33,23,2423,33,23,1,7,6,8787,5,33,2,3,-23,-54,-67,100,400};
vector ret=twoSum7(nums, 393);
for (int i:ret) cout<
为什么同向双指针可以work? 看下面这张图就知道了。P2就相当于Y,P1就相当于X。最开始坐标在(0,1),蓝色图标箭头显示了该算法是如何一步一步走到正确位置的。看懂了就可以明白为什么p2-p1>target时是p1++而不是p2–, p2-p1
同向双指针的另外一种写法。
struct ResultType {
int value;
int index;
ResultType(int v = 0, int i = 0) : value(v), index(i) {}
};
struct cmp {
bool operator() (ResultType & a, ResultType & b) {
if (a.value < b.value) return true;
if (a.value == b.value) return a.index < b.index;
return false;
}
} compare;
class Solution {
public:
/**
* @param nums: an array of Integer
* @param target: an integer
* @return: [index1 + 1, index2 + 1] (index1 < index2)
*/
vector twoSum7(vector &nums, int target) {
int n = nums.size();
if (n == 0) return vector();
vector nums2;
for (int i = 0; i < n; ++i) {
nums2.push_back(ResultType(nums[i], i));
}
sort(nums2.begin(), nums2.end(), compare);
int p1 = 0, p2 = 1;
if (target < 0) target = -target;
while(p2 < n) {
int diff = nums2[p2].value - nums2[p1].value;
if (diff == target) {
return vector{min(nums2[p1].index, nums2[p2].index) + 1, max(nums2[p1].index, nums2[p2].index) + 1};
} else if (diff < target) {
p2++;
} else {
p1++;
if (p1 == p2) p2++;
}
}
return vector();
}
};
解法2:Hash Table (unordered_set或unordered_map)
vector twoSum7(vector &nums, int target) {
unordered_map hashMap;
vector result;
for (int i=0; i