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;
}
};