Update single-number-iii.py

kamyu104 · kamyu104 · commit 61250cfecdfd · 2015-09-03T00:30:37.000+08:00
diff --git a/Python/single-number-iii.py b/Python/single-number-iii.py
@@ -17,6 +17,17 @@
 #
 
 class Solution:
+    # @param {integer[]} nums
+    # @return {integer[]}
+    def singleNumber(self, nums):
+        x_xor_y = reduce(operator.xor, nums)
+        x_xor_y &= -x_xor_y
+        result = [0, 0]
+        for i in nums:
+            result[not i & x_xor_y] ^= i
+        return result
+        
+class Solution2:
     # @param {integer[]} nums
     # @return {integer[]}
     def singleNumber(self, nums):
@@ -31,4 +42,4 @@ def singleNumber(self, nums):
             if i & bit:
                 x ^= i
 
-        return [x, x_xor_y ^ x]
+        return [x, x ^ x_xor_y]
