利用鸽巢原理检测顺序数组中一个数字在或不在
问题描述:
问题分析:
鸽巢原理(组合数学知识):
n个巢, n+1只鸽子,每个鸽子进一个巢,那种总会剩下一个鸽子无家可归;
在此问题中我们假设数字的下标为鸽巢,下标对应的值为鸽子编号。
经过一次遍历让鸽子(回到鸽子编号-1的巢里)回家,最终发现无家可归的鸽子,和没有鸽子的巢。
更详细的讲解请查看:
鸽巢思想解决
完整代码:
// 检测缺失数字
for (int i = 0; i < n; i++)
nums[abs(nums[i]) - 1] = 0-abs(nums[abs(nums[i])-1]);
for (int i = 0; i < n; i++)
{
if (nums[i] > 0)
res.push_back(i+1);
}
//检测重复数字
for (int i = 0; i < nums.size(); i++)
{
if (nums[abs(nums[i]) - 1] < 0)
res.push_back(abs(nums[i]));
else
nums[abs(nums[i]) - 1] = -nums[abs(nums[i])-1];
}
//其中 nums[i]-1为标号nums[i]的鸽子本应该去的巢;添加abs保证下标计算正确,(因为有可能这个巢里已经有鸽子了)
//将编号X鸽子本应该去的巢里置为负数,表示此巢已有鸽子;
//为了后续计算正确,仍应保证索去巢里的值不变,通过将目标巢里原有的鸽子编号取负来保证错误鸽子可以回到原有巢中
//以上述案例4 3 2 7 8 2 3 1为例
//i= 0时鸽子编号是4,他应该去3号巢(差1的原因是数组下标从0开始),所以下标num[i]-1 = 3对应的值置负,表示已有鸽子归巢;
//等遍历至下标i= 3时 鸽子编号为nums[3] = -7,负号表示已有鸽子回来,7表示7号鸽子还没回家,所以下标abs(nums[3])-1 = 6对应值置负,表示7号鸽子归巢;
// 所以在后续利用中,巢内为负表示已有鸽子回来,若一个鸽子映射到为负值的巢内时,说明这只鸽子重复,若遍历完了之后,仍有巢为正,则表示该巢对应的鸽子缺失;