剑指 Offer 03. 数组中重复的数字 - 力扣(LeetCode) (leetcode-cn.com)
方案1
拿空间换时间,时间复杂度O( n ),空间复杂度O( n )
运行结果
代码
class Solution {
public:
int findRepeatNumber(vector<int>& nums) {
int* nMap = new int[nums.size()]();
for (auto x : nums) if (++nMap[x] > 1) return x;
return -1;
}
};
方案2
注意这个数组元素的特点:从0到n-1。如果没有重复的数字的话,把它排好序之后,第 i 个位置上的数字应该正好是 i。在有重复数字的情况下,如果我们每碰到一个数字x,就让它回到原位(第x个位置)的话,最终一定会出现这种情况:第 i 个位置上的数字是m,但第m个位置上的数字也是m,这时,就找到了一个重复数字。根据这个原理,我们可以设计出来一种时间复杂度O( n ),空间复杂度为O( 1 )的算法:
从0开始遍历数组,如果 i 个位置上的数字是 i 的话,那么检查下一位;如果第 i 个位置上的数字不是 i的话(不妨设它为m),那就去比较m和第m个位置上的数字,如果相等,那么就找到了一个重复数字,如果不等,那么把第 i 个位置上的数字m个第m个位置上的数字相交换,即使m回到原位置。如此循环进行。
运行结果
代码
class Solution {
public:
int findRepeatNumber(vector<int>& nums) {
int n = nums.size();
int i = 0;
while (i != n) {
if (nums[i] != i) {
if (nums[i] == nums[nums[i]]) return nums[i];
else swap(nums[i], nums[nums[i]]);
}
else ++i;
}
return -1;
}
};