程序员面试金典 16.24

Pairs with Sum:找出数组中和为给定值target的所有数对,每个数只能使用一次。

看这题的数据量,暴力枚举法肯定是超时的,所以只能去优化内层循环,也就是用map或者哈希。可以先使用一个map来记录每个数字的原始数量,这样匹配的时候,只需查看该数字是否有剩余的即可。

这里需要注意类似[6]target = 12的测试用例,在这种情况下结果应该是空。

class Solution {
public:
    vector<vector<int>> pairSums(vector<int>& nums, int target) {
        vector<vector<int>> ret;
        map<int, size_t> Remain;
        for_each(nums.begin(), nums.end(),
            [&Remain](const int i)
            {
                Remain[i]++;
            });
        for(auto i : nums)
        {
            size_t &cnt1 = Remain[i], &cnt2 = Remain[target - i];
            if(cnt1 != 0){
                cnt1--;
                if(cnt2 != 0){
                    ret.emplace_back(1, i);
                    ret.back().push_back(target - i);
                    cnt2--;
                }
                else cnt1++;
            }
        }
        return ret;
    }
};

也可以将数组排序,然后使用双指针的方式,还是要特别注意处理两个指针指向同一个元素的情况。

class Solution {
public:
    vector<vector<int>> pairSums(vector<int>& nums, int target) {
        sort(nums.begin(), nums.end());
        vector<vector<int>> ret;
        auto iter = nums.begin();
        auto riter = nums.rbegin();
        while(iter + 1 < riter.base()){
            int sum = *iter + *riter;
            if(sum < target) iter++;
            else if(sum > target) riter++;
            else{
                ret.push_back(vector<int>());
                ret.back().push_back(*iter);
                ret.back().push_back(*riter);
                iter++, riter++;
            }
        }
        return ret;
    }
};

你可能感兴趣的:(《程序员面试金典》)