Leetcode 397.整数替换（Integer Replacement）

Leetcode 397.整数替换

1 题目描述（Leetcode题目链接）

  给定一个正整数 n，你可以做如下操作：

1. 如果 n 是偶数，则用 n / 2替换 n。
2. 如果 n 是奇数，则可以用 n + 1或n - 1替换 n。

n 变为 1 所需的最小替换次数是多少？

输入:
8

输出:
3

解释:
8 -> 4 -> 2 -> 1

输入:
7

输出:
4

解释:
7 -> 8 -> 4 -> 2 -> 1
或
7 -> 6 -> 3 -> 2 -> 1

2 题解

  看到这个问题第一个想法就是递归解决，这样既好想又好写。

class Solution:
    def integerReplacement(self, n: int) -> int:
        if n == 1:
            return 0
        if n % 2 == 0:
            return 1 + self.integerReplacement(n//2)
        else:
            return 1 + min(self.integerReplacement(n+1), self.integerReplacement(n-1))

但是递归解决这个问题的效率并不高。可以使用位运算的方法来分析这个问题，在奇数的情况下：

• 如果n的二进制后两位是01，则n减1；
• 如果n的二进制后两位是11，则n加1；
• 特殊情况，如果n等于3，那么应该减1；

class Solution:
    def integerReplacement(self, n: int) -> int:
        retv = 0
        while n != 1:
            if n % 2 == 0:
                n //= 2
            else:
                if n & 2 == 0 or n ==3:
                    n -= 1
                else:
                    n += 1
            retv += 1
        return retv