题目描述:
一个整型数组 nums 里除两个数字之外,其他数字都出现了两次。请写程序找出这两个只出现一次的数字。要求时间复杂度是O(n),空间复杂度是O(1)。
示例 1:
输入:nums = [4,1,4,6]
输出:[1,6] 或 [6,1]
示例 2:
输入:nums = [1,2,10,4,1,4,3,3]
输出:[2,10] 或 [10,2]
*限制:2 <= nums <= 10000
先考虑上述问题的简单版本:一个数组里面只有一个数字出现一次,其他都出现两次,请找出这个数字。
这个问题可以可以使用用异或的性质解决。异或的性质:对于整数a,有
(1) a^a=0
(2)a^0=a
(2)a^b^c=a^(b^c)=(a^c)^b
利用以上的性质,上面的题目的解法为:
func singleNumbers(nums []int) int {
xorSum := 0
for _, v := range nums {
xorSum ^= v
}
return xorSum
}
有了以上的基础,解文章开始时提出的问题的思路如下:
(1)对于出现两次的元素,使用“异或”操作后结果肯定为0,那么我们就可以遍历一遍数组,对所有元素使用异或操作,那么得到的结果就是两个出现一次的元素的异或结果。
(2)因为这两个元素不相等,所以异或的结果肯定不是0,也就是可以再异或的结果中找到1位不为0的位,例如异或结果的最后一位不为0。
(3)这样我们就可以最后一位将原数组元素分为两组,一组该位全为1,另一组该位全为0。
(4)再次遍历原数组,最后一位为0的一起异或,最后一位为1的一起异或,两组异或的结果分别对应着两个结果。
复杂度:
(1)时间复杂度:第一次循环,将所有元素异或得到对应结果,时间开销为O(n);第二次循环,找出第一次异或结果为1的位,时间开销为O(32);第三次循环,根据为1的位将元素分为两组进行异或得到两个结果,时间复杂度为O(n),所以总的时间复杂度为T(n) = 2*O(n)+O(32) = O(n)。
(2)空间复杂度:常数,因为只分配了两个空间用于结果的保存,因此空间复杂度为常数。
func singleNumbers(nums []int) []int {
tar := make([]int, 2)
xorSum := 0
for _, v := range nums {
xorSum ^= v
}
div := 1
for ; div & xorSum == 0; {
div <<= 1
}
for _, v := range nums {
if div & v == 0 {
tar[0] ^= v
} else {
tar[1] ^= v
}
}
return tar
}