LintCode 610: Two Sum - Difference equals to target (经典题:两数差等于给定值)

解法1:先排序后用双指针。这里用逆向双指针好像不太可行,因为假设target=6, 两个符合条件的数的索引可能是(1,7), (3,9), (7,12),等等。假如我们设p1=0, p2=12,但是nums[p2]-nums[p1]>6,那我们是p1++还是p2–呢? 两个都进行就复杂了。
这里用同向双指针比较好。一开始p1=0, p2=1。当两指针指向的值的differencetarget时,p1++(但这里要注意p1不能超过p2,若p1==p2,则p2++),若difference=target,则返回p1,p2。
这里要先用一个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 LintCode 610: Two Sum - Difference equals to target (经典题:两数差等于给定值)_第1张图片
同向双指针的另外一种写法。

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

你可能感兴趣的:(LintCode 610: Two Sum - Difference equals to target (经典题:两数差等于给定值))