From 550458b5ef4a31e8c95e4fcd0f22bc6ca59fd46e Mon Sep 17 00:00:00 2001 From: Jack Sangdahl Date: Tue, 11 Mar 2025 11:38:29 -0600 Subject: 397. integer replacement --- integer_replacement.py | 33 +++++++++++++++++++++++++++++++++ 1 file changed, 33 insertions(+) create mode 100644 integer_replacement.py (limited to 'integer_replacement.py') diff --git a/integer_replacement.py b/integer_replacement.py new file mode 100644 index 0000000..8d7dbda --- /dev/null +++ b/integer_replacement.py @@ -0,0 +1,33 @@ +# 397. Integer Replacement + +""" +given a positive integer 'n', you can apply one of the following operations: +if 'n' is even, replace 'n' witn 'n / 2'. if 'n' is odd, replace 'n' with +either 'n + 1' or 'n - 1'. return the minimum number of operations needed for +'n' to become 1 +""" + + +class Solution(object): + def integerReplacement(self, n): + """ + :type n: int + :rtype: int + """ + ans = 0 + while n > 1: + ans += 1 + if n % 2 == 0: + n //= 2 + elif n % 4 == 1 or n == 3: + n -= 1 + else: + n += 1 + return ans + + +if __name__ == "__main__": + obj = Solution() + print(obj.integerReplacement(n=8)) + print(obj.integerReplacement(n=7)) + print(obj.integerReplacement(n=4)) -- cgit v1.2.3