EPI: closest int, same weight (py)

jsgoller1 · jsgoller1 · commit a86f247caebb · 2023-08-02T21:42:47.000-04:00
diff --git a/elements-of-programming-interviews/problem_mapping.js b/elements-of-programming-interviews/problem_mapping.js
@@ -66,7 +66,7 @@ problem_mapping = {
                 "total": 10000
             },
             "Python: closest_int_same_weight.py": {
-                "passed": 0,
+                "passed": 10000,
                 "total": 10000
             }
         },
diff --git a/elements-of-programming-interviews/python/closest_int_same_weight.py b/elements-of-programming-interviews/python/closest_int_same_weight.py
@@ -2,8 +2,18 @@
 
 
 def closest_int_same_bit_count(x: int) -> int:
-    # TODO - you fill in here.
-    return 0
+    original = x
+    i = 0
+    curr = prev = x & 1 != 0
+
+    while (prev == curr):
+        x >>= 1
+        prev = curr
+        curr = x & 1 != 0
+        i += 1
+
+    mask = (1 << i) | (1 << i-1)
+    return original ^ mask
 
 
 if __name__ == '__main__':

Original file line number	Diff line number	Diff line change
`@@ -66,7 +66,7 @@ problem_mapping = {`
`66`	`66`	`"total": 10000`
`67`	`67`	`},`
`68`	`68`	`"Python: closest_int_same_weight.py": {`
`69`		`- "passed": 0,`
	`69`	`+ "passed": 10000,`
`70`	`70`	`"total": 10000`
`71`	`71`	`}`
`72`	`72`	`},`