Update Single Number III.java

varunu28 · web-flow · commit d180b8001c61 · 2021-11-05T17:16:40.000-07:00
diff --git a/Medium/Single Number III.java b/Medium/Single Number III.java
@@ -1,15 +1,17 @@
 class Solution {
   public int[] singleNumber(int[] nums) {
-    Set<Integer> set = new HashSet<>();
+    int xorValue = Integer.highestOneBit(Arrays.stream(nums)
+        .boxed()
+        .reduce((a, b) -> a ^ b)
+        .orElse(0));
+    int[] result = new int[2];
     for (int num : nums) {
-      if (set.contains(num)) {
-        set.remove(num);
-      }
-      else {
-        set.add(num);
+      if ((xorValue & num) == 0) {
+        result[0] ^= num;
+      } else {
+        result[1] ^= num;
       }
     }
-    Iterator<Integer> iter = set.iterator();
-    return new int[]{iter.next(), iter.next()};
+    return result;
   }
 }
