397.整数替换

题目描述

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

如果 n 是偶数，则用 n / 2替换 n 。
如果 n 是奇数，则可以用 n + 1或n - 1替换 n 。
n 变为 1 所需的最小替换次数是多少？

链接：https://leetcode-cn.com/problems/integer-replacement

分析

很明显的，按照题目所给的意思我们可以很容易地写出递归地代码
但递归的最大缺点在于，有多个子问题被重复的调用了
比如说n=3,
计算它时会递归进入n=4和n=2，而n=4里面又需要n=2的答案
所以我们就需要用某种手段记录我们已经计算过的值
同时还要能迅速地查到值所对应需要的次数
很明显的，就是用哈希表记录啦
这种叫做记忆化搜索，它和动态规划高效的原因都是因为有记录子问题的解，解决了子问题的重复带来的复杂度问题，

代码

class Solution:
    def integerReplacement(self, n: int) -> int:
        hashmap = {}
        def dfs(n):
            if n in hashmap:
                return hashmap[n]
            if n == 1:
                return 0
            if n %2 == 0:
                hashmap[n] = dfs(n/2)+1
                return dfs(n/2)+1
            else:
                hashmap[n] = min(dfs(n+1),dfs(n-1))+1
                return min(dfs(n+1),dfs(n-1))+1
        return dfs(n)

不过后来看题解，发现原来还有@cache的装饰器可以直接拿来用

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

后面学习到cache再来吧2333