在一个长度为 n 的数组 nums 里的所有数字都在 0~n-1 的范围内。数组中某些数字是重复的,但不知道有几个数字重复了,也不知道每个数字重复了几次。请找出数组中任意一个重复的数字。
方法一:依次将数字存入集合,存入失败则说明有重复元素
// c++
#include <vector>
#include <set>
using namespace std;
class Solution
{
public:
int findRepeatNumber(vector<int> &nums)
{
int repeat = -1;
set<int> hashset;
for (int num : nums)
{
if(!hashset.insert(num).second){
repeat = num;
break;
}
}
return repeat;
}
};
注意点:
- 忽略了题干中数组元素为0~n-1的重要条件,因此该算法占用的空间大。
- set模板的使用,它的insert函数将num插入set中,并且返回一个pair,如果插入失败,则返回的pair的第二个元素为false。
// java
import java.util.HashSet;
public class Solution {
public int findRepeatNumber(int[] nums) {
int repeat = -1;
HashSet<Integer> set = new HashSet<Integer>();
for (int num : nums) {
if (!set.add(num)) {
repeat = num;
break;
}
}
return repeat;
}
}
注意点:
- 生成一个HashSet: HashSet<Integer> set = new HashSet<Integer>();
- set.add(e)将e插入set,如果已经有e,则返回false。
# python
class Solution(object):
def findRepeatNumber(self, nums):
repeat = -1
dic = set()
for num in nums:
if num in dic:
repeat = num
break
dic.add(num)
return repeat
注意点:
- 没什么好说的,python的代码就是简单明了。
方法二:原地交换法
// c++
#include <vector>
using namespace std;
class Solution
{
public:
int findRepeatNumber(vector<int> &nums)
{
int repeat = -1;
for (int i = 0; i < nums.size(); i++)
{
int m = nums[i];
if(m == i){
continue;
}
if(nums[m] == m){
repeat = m;
break;
}
nums[i] = nums[m];
nums[m] = m;
i--;
}
return repeat;
}
};
注意点:
- 遍历数组,将每个萝卜放到它应该放入的坑中。空间复杂度为O(1)。
// java
public class Solution {
public int findRepeatNumber(int[] nums) {
int i = 0;
while (i < nums.length) {
if (nums[i] == i) {
i++;
continue;
}
if (nums[i] == nums[nums[i]]) {
return nums[i];
}
int temp = nums[i];
nums[i] = nums[temp];
nums[temp] = temp;
}
return -1;
}
}
注意点:
- 用while遍历比用for遍历要好。
- 对于有返回值的函数,先写好默认返回值。
# python
class Solution(object):
def findRepeatNumber(self, nums):
i = 0
while i < len(nums):
m = nums[i]
if m == i:
i += 1
continue
if nums[m] == m: return m
nums[i], nums[m] = nums[m], nums[i]
return -1
注意点:
- python中两个数字的交换可以直接写a, b = b, a