有序数组中不同平方值的个数(头条面试题)---C++实现

题目：

求解一个有序数组不同平方值的个数，{-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;
}