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