题目:
求解一个有序数组不同平方值的个数,{-1,-1,0,1,1}这里平方值只有0,1两种情况,返回2;{-3,-1,0,0,2,3}这里拼房族有0,1,4,9共4中情况,返回4。
要求:时间复杂度O(n)空间复杂度O(1)
思路:
刚开始看到这题,哈希表应该是大多数人都能想到的,且复杂度要求都能满足。就是遍历数组,把每个数的平方放入哈希表中去重即可。
但是这种想法很难通过面试,因为有序的条件没有利用到。
正解:
有序数组普遍是利用双指针问题求解,这里也一样。每次检查头尾指针的和是等于0、大于0还是小于0.根据这3中情况分析。注意数字可能重复,需要去重。
int diffSquareNum(const vector<int>& nums) {
if (!nums.size())
return 0;
int left = 0;
int right = nums.size() - 1;
int sum = 0;
while (left <= right) {
if (nums[left] + nums[right] == 0) {
sum++;
int t = nums[left];
//这里是为了去重
while (left <= right && nums[left] == t)
left++;
while (left <= right && nums[right] == -t)
right--;
}
else if (nums[left] + nums[right] < 0) {
sum++;
int t = nums[left];
while (left <= right && nums[left] == t)
left++;
}
else{
sum++;
int t = nums[right];
while (left <= right && nums[right] == t)
right--;
}
}
return sum;
}