在刷LeetCode-15三数之和发现了一个vector.size()的返回值size_t类型可能引发的问题,之前看到过有人规避过这个问题单一直没有亲身碰到过也就不以为然,今天终于见识了,LeetCode调试器调试了半天才发现问题所在- - !。
首先是有问题的代码
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> res;
if(nums.empty()) return res;
// 排序一下在做
sort(nums.begin(), nums.end());
if(nums[0]>0 || nums[nums.size()-1]<0) return res;
// i<nums.size()-2是因为最后要留两个数来相加
for(int i=0; i<(nums.size()-2; ++i){ // ### 1
if(nums[i]>0) break; // 第一个元素为正数之后就不需要再遍历了
if(i>0 && nums[i]==nums[i-1]) continue; // 避免找到重复的
int left=i+1, right=nums.size()-1;
while(left<right&&nums[right]>=0){
if(nums[left]+nums[right] == -nums[i]){
res.push_back({nums[i], nums[left], nums[right]});
// 更新位置,跳过重复的数
while(left<right && nums[left]==nums[left+1]) left++;
while(left<right && nums[right]==nums[right-1]) right--;
left++, right--;
}
else if(nums[left]+nums[right] < -nums[i]){
left++;
}
else{
right--;
}
}
}
return res;
}
};
问题发生在for循环的条件处(###1),判断条件为i<nums.size()-2
,当nums.size()-x>=0
时不会出现问题,但当nums.size()-x<0
问题就发生了,由于nums.size()
的返回值类型为size_t
是一个无符号类型整数,因而当nums.size()-x<0
时会发生无符号整数的下溢,从而变成一个很大的正整数,也就使得循环成为了一个“死循环”。
解决方案:在nums.size()
前加上一个强制类型转换,将size_t
类型转换为int
类型,这样就会得到负数,从而不会进入循环,正确代码如下:
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> res;
if(nums.empty()) return res;
// 排序一下在做
sort(nums.begin(), nums.end());
if(nums[0]>0 || nums[nums.size()-1]<0) return res;
// i<nums.size()-2是因为最后要留两个数来相加
// nums.size()是size_t无符号整型,当-2为负数后溢出后变为正整数上限
for(int i=0; i<(int)nums.size()-2; ++i){
if(nums[i]>0) break; // 第一个元素为正数之后就不需要再遍历了
if(i>0 && nums[i]==nums[i-1]) continue; // 避免找到重复的
int left=i+1, right=nums.size()-1;
while(left<right&&nums[right]>=0){
if(nums[left]+nums[right] == -nums[i]){
res.push_back({nums[i], nums[left], nums[right]});
// 更新位置,跳过重复的数
while(left<right && nums[left]==nums[left+1]) left++;
while(left<right && nums[right]==nums[right-1]) right--;
left++, right--;
}
else if(nums[left]+nums[right] < -nums[i]){
left++;
}
else{
right--;
}
}
}
return res;
}
};