- 寻找重复数
给定一个包含 n + 1 个整数的数组 nums,其数字都在 1 到 n 之间(包括 1 和 n),可知至少存在一个重复的整数。假设只有一个重复的整数,找出这个重复的数。
示例 1:
输入: [1,3,4,2,2]
输出: 2
示例 2:
输入: [3,1,3,4,2]
输出: 3
说明:
不能更改原数组(假设数组是只读的)。
只能使用额外的 O(1) 的空间。
时间复杂度小于 O(n2) 。
数组中只有一个重复的数字,但它可能不止重复出现一次。
思路1:用set集合作弊(违反了限制2)
class Solution {
public:
int findDuplicate(vector<int>& nums) {
unordered_set<int> s;
int digit;
for (int &num : nums) {
if (!s.empty() && s.count(num)) {
digit = num;
break;
}
s.insert(num);
}
return digit;
}
};
思路2:快速排序(违反了限制1)
class Solution {
public:
void quickSort(vector<int>& nums, int left, int right) {
if (left >= right) return;
int pivot = (left + right) / 2;
int tmp, tbegin = left;
vector<int> less;
vector<int> greater;
for (int i = left; i <= right; i ++) {
if (i == pivot) tmp = nums[pivot];
else if (nums[i] <= nums[pivot]) less.push_back(nums[i]);
else greater.push_back(nums[i]);
}
for (int i = 0; i < less.size(); i ++) nums[tbegin ++] = less[i];
nums[tbegin ++] = tmp; pivot = tbegin - 1;
for (int i = 0; i < greater.size(); i ++) nums[tbegin ++] = greater[i];
quickSort(nums, left, pivot - 1);
quickSort(nums, pivot + 1, right);
}
int findDuplicate(vector<int>& nums) {
quickSort(nums, 0, nums.size() - 1);
int digit;
for (int i = 0; i < nums.size() - 1; i ++) {
if (nums[i] == nums[i + 1]) {
digit = nums[i];
}
}
return digit;
}
};
可以用c++自带的排序函数
class Solution {
public:
int findDuplicate(vector<int>& nums) {
sort(nums.begin(), nums.end());
int digit;
for (int i = 0; i < nums.size() - 1; i ++) {
if (nums[i] == nums[ i + 1]) {
digit = nums[i];
break;
}
}
return digit;
}
};
思路3:抽屉原理 + 二分
用二分法确定范围。
抽屉原理:桌上有十个苹果,要把这十个苹果放到九个抽屉里,无论怎样放,我们会发现至少会有一个抽屉里面放不少于两个苹果。
根据题目条件,在【left, right】这些抽屉里,每个抽屉放和他相同的数,则必然有一个抽屉放了两个重复数。如果申请一个抽屉数组去做,将不符合限制2。因此引申到了二分法的另一个应用:确定范围。
在【left, right】这些抽屉里找一个中间抽屉mid,向【left, mid】抽屉里放数。统计数组元素<=mid的个数。若个数大于mid,则重复数一定在[left, mid]区间里,反之则在[mid + 1, right]区间里。
class Solution {
public:
int findDuplicate(vector<int>& nums) {
int left = 1, right = nums.size() - 1, cnt = 0;
while (left < right) {
int mid = (left + right) / 2;
for (int i = 0; i < nums.size(); i ++) {
if (nums[i] <= mid) cnt ++;
}
if (cnt > mid) right = mid;
else left = mid + 1;
cnt = 0;
}
return left;
}
};