go: add «260. Single Number III»
Signed-off-by: Matej Focko <me@mfocko.xyz>
This commit is contained in:
parent
27d260c9c8
commit
c66b1879eb
1 changed files with 33 additions and 0 deletions
33
go/single-number-iii.go
Normal file
33
go/single-number-iii.go
Normal file
|
@ -0,0 +1,33 @@
|
||||||
|
package single_number_iii
|
||||||
|
|
||||||
|
func singleNumber(nums []int) []int {
|
||||||
|
reduce := func() int {
|
||||||
|
xor := 0
|
||||||
|
for _, x := range nums {
|
||||||
|
xor ^= x
|
||||||
|
}
|
||||||
|
return xor
|
||||||
|
}
|
||||||
|
xor := reduce()
|
||||||
|
|
||||||
|
findMask := func() int {
|
||||||
|
mask := 1
|
||||||
|
for (xor & mask) == 0 {
|
||||||
|
mask <<= 1
|
||||||
|
}
|
||||||
|
return mask
|
||||||
|
}
|
||||||
|
mask := findMask()
|
||||||
|
|
||||||
|
a := 0
|
||||||
|
b := 0
|
||||||
|
for _, x := range nums {
|
||||||
|
if (x & mask) != 0 {
|
||||||
|
a ^= x
|
||||||
|
} else {
|
||||||
|
b ^= x
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return []int{a, b}
|
||||||
|
}
|
Loading…
Reference in a new issue