442. 数组中重复的数据 ( 抽屉原理 )
程序员文章站
2022-03-16 11:04:45
...
LeetCode: 442. 数组中重复的数据
核心条件 : 其中1 ≤ a[i] ≤ n (n为数组长度)
思路分析:“桶排序”的思想是“抽屉原理”,即“一个萝卜一个坑”,8 个萝卜要放在 7 个坑里,则至少有 1 个坑里至少有 2 个萝卜。
这里由于数组元素限定在数组长度的范围内,因此,我们可以通过一次遍历:
让数值 1 就放在索引位置 0 处;
让数值 2 就放在索引位置 1 处;
让数值 3 就放在索引位置 2 处;
……
一次遍历以后,那些“无处安放”的元素就是我们要找的“出现两次的元素”。
抽屉原理
func findDuplicates(nums []int) []int {
// 抽屉原理
var ans []int
// 核心代码
for i, _ := range nums{
for nums[nums[i] - 1] != nums[i] {
swap(nums, nums[i] - 1, i)
// fmt.Println(nums)
}
}
for i, val := range nums {
if nums[i] != i + 1 {
ans = append(ans, val)
}
}
return ans
}
func swap(arr []int, a int, b int){
if a == b { return }
arr[a] = arr[a] ^ arr[b]
arr[b] = arr[a] ^ arr[b]
arr[a] = arr[a] ^ arr[b]
}
参考–weiwei哥题解: “抽屉原理” + 基于“异或运算”交换两个变量的值
上一篇: UDP实现通信