【剑指】53(2).0到n-1中缺失的数字

题目描述

• 一个长度为n-1的递增排序数组中的所有数字都是唯一的，并且每个数字都在范围0到n-1之内。在范围0到n-1的n个数字中有且只有一个数字不在该数组中，请找出这个数字。

算法分析

• 若数组没有缺失，则每个数字和它的下标都相等。然而，现在数组有缺失，说明从缺失的那个数开始，后面的数字都比它的下标大1。因此找出第一个数值和下标不相等的数，那么，它的下标就是缺失的那个数。 数组可看成两部分，前半段数值和下标相等，后半段数值和下标不相等，且数组是有序的。所以，使用二分查找，若中间数的值和下标相等，则在后半段继续寻找；若中间数的值和下标不相等，且中间数的前一个元素的值等于它的下标相等，那么这个中间数就是第一个值和下标不相等的数，它的下标就是缺失的那个数，否则继续在前半段寻找。

提交代码：

class Solution {
public:
	int GetMissingNumber(vector<int> numbers)
	{
		if (numbers.empty())
			return -1;

		int start = 0;
		int end = numbers.size() - 1;

		while (start <= end)
		{
			int middle = (start + end) >> 1;

			if (numbers[middle] == middle)
				start = middle + 1;
			else
			{
				if (middle == 0 || numbers[middle - 1] == (middle - 1))
					return middle;
				end = middle - 1;
			}
				
		}

		if (start == numbers.size())
			return start;

		return -1;
	}
};

测试代码：

// ====================测试代码====================
void Test(const char* testName, vector<int> numbers, int expected)
{
	if (testName != nullptr)
		printf("%s begins: ", testName);
	Solution s;
	int result = s.GetMissingNumber(numbers);
	if (result == expected)
		printf("Passed.\n");
	else
		printf("Failed.\n");
}

// 缺失的是第一个数字0
void Test1()
{
	vector<int> numbers = { 1, 2, 3, 4, 5 };
	int expected = 0;
	Test("Test1", numbers, expected);
}

// 缺失的是最后一个数字
void Test2()
{
	vector<int> numbers = { 0, 1, 2, 3, 4 };
	int expected = 5;
	Test("Test2", numbers, expected);
}

// 缺失的是中间某个数字0
void Test3()
{
	vector<int> numbers = { 0, 1, 2, 4, 5 };
	int expected = 3;
	Test("Test3", numbers, expected);
}

// 数组中只有一个数字，缺失的是第一个数字0
void Test4()
{
	vector<int> numbers = { 1 };
	int expected = 0;
	Test("Test4", numbers, expected);
}

// 数组中只有一个数字，缺失的是最后一个数字1
void Test5()
{
	vector<int> numbers = { 0 };
	int expected = 1;
	Test("Test5", numbers, expected);
}

// 空数组
void Test6()
{
	int expected = -1;
	Test("Test6", vector<int>(), expected);
}

int main(int argc, char* argv[])
{
	Test1();
	Test2();
	Test3();
	Test4();
	Test5();
	Test6();

	return 0;
}