这是一道微软面试题
题目:
分析:
假设不管毒药和水混合,不管毒药浓度为多少,小鼠喝过就会死。
现进行这样的实验(原理二分法):1000瓶水,先将前500的混合,让一只小鼠喝。
如果死亡,则前500有毒药,后500瓶直接排除;若不死亡,则后500瓶有毒药。下次将进行250瓶的混合,并试毒选择。
以此类推,二分结束的条件是begin = end 最后直接输出begin即毒药的标签号。题目数据设置的也比较巧妙, 2^10 = 1024 > 1000,因此10只老鼠可以满足试毒要求。
问题的时间复杂度O(log n), 空间复杂度O(1)。
解答:
#include <iostream>
#include <vector>
using namespace std;
int selectPoison(vector<bool>& bottle) {
int begin = 0, end = bottle.size() - 1;
while (begin < end) {
int middle = (begin + end) / 2;
bool ispoison = false;
for (int i = begin; i <= middle; ++i) {
ispoison ^= bottle[i];//这个过程模拟毒药的混合
if (ispoison) break;
}
if (ispoison) end = middle;
else begin = middle + 1;
}
return begin;
}
int main() {
vector<bool> bottle(1000, false);
bottle[403] = true;//举个例子,有毒为true
cout<< selectPoison(bottle)<< endl;
return 0;
}
如有错误或者更好的解法,欢迎大家批评指正!