問題描述:
給定一個包含 n + 1 個整數的數組 nums,其數字都在 1 到 n 之間(包括 1 和 n),可知至少存在一個重複的整數。假設只有一個重複的整數,找出這個重複的數
來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/find-the-duplicate-number
示例 1:
輸入: [1,3,4,2,2]
輸出: 2
示例 2:
輸入: [3,1,3,4,2]
輸出: 3
說明:
不能更改原數組(假設數組是隻讀的)。
只能使用額外的 O(1) 的空間。
時間複雜度小於 O(n2) 。
數組中只有一個重複的數字,但它可能不止重複出現一次。
解題思路:
已知題目要求,時間複雜度和只讀要求,且只有一個重複數字
可以使用二分法
比如:
[1,3,4,2,2] left=1,right=5;循環條件(left<right)
mid = 3; 數組中小於等於3的數目爲4>3;此時 將right=mid=3;
mid = 2; 數組中小於等於2的數目爲3>2;此時 right = 2;
mid = 1; 數組中小於等於1的數目爲1==1,此時 left = 2
終止循環 return right
完整代碼如下 :
class Solution {
public:
int findDuplicate(vector<int>& nums) {
int left=1,right = nums.size();
while(left<right){
int mid = left + (right-left)/2;
int cnt=0;
for(auto val:nums){
if(val<=mid){
cnt++;
}
}
if(cnt>mid) right=mid;
else left = mid+1;
}
return right;
}
};
還有一種思路是快慢指針的解法:
class Solution {
public:
int findDuplicate(vector<int>& nums) {
int fast = 0,slow=0;
while(true){
fast = nums[nums[fast]];
slow = nums[slow];
if(slow == fast){
fast=0;
while(nums[slow]!=nums[fast]){
fast = nums[fast];
slow = nums[slow];
}
return nums[slow];
}
}
}
};