leetcode 397. 整数替换(记忆化搜索)

最新推荐文章于 2022-12-07 15:05:37 发布

不想在山底徘徊的小蜗牛

最新推荐文章于 2022-12-07 15:05:37 发布

阅读量564

点赞数 1

分类专栏：递归文章标签： leetcode 算法 dfs

本文链接：https://blog.csdn.net/weixin_42051691/article/details/124211110

版权

递归专栏收录该内容

9 篇文章 1 订阅

订阅专栏

该博客介绍了一个使用深度优先搜索（DFS）解决LeetCode第397题的问题，即找到使整数n变为1所需的最小替换次数。通过递归和记忆化优化，当n为偶数时除以2，为奇数时选择加1或减1，找到最短路径。代码中使用了unordered_map进行记忆化存储，避免重复计算。

摘要由CSDN通过智能技术生成

397. 整数替换

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

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

示例 1：

输入：n = 8 输出：3 解释：8 -> 4 -> 2 -> 1

示例 2：

输入：n = 7 输出：4 解释：7 -> 8 -> 4 -> 2 -> 1 或 7 -> 6 -> 3 -> 2 -> 1

示例 3：

输入：n = 4 输出：2

提示：

1 <= n <= 231 - 1

解析：
可以使用深度优先搜索：
递归出口：n==1,返回值为0
递归过程：
1, n为偶数，直接递归下一项 n/2，步数+1
2，n为奇数，递归下一项 n+1，n-1，步数+1，取两者最小值
然后用记忆化优化，可以使用map，用int开的空间过大，会有很多用不到的空间。
code：

class Solution {
public:
    unordered_map<int,int> ma;
    int dfs(long n){
        if(ma.find(n)!=ma.end())
            return ma[n];
        int ans;
        if(n==1){
            ma[n]=0;
            return 0;
        }
            
        if(n%2==0){
            int tmp=dfs(n/2)+1;
            ma[n]= tmp;
            return ma[n];
        }
            
        else if(n%2!=0){
            int tmp=min(dfs(n-1),dfs(n+1))+1;
            ma[n]=tmp;
            return ma[n];
        }
        return ma[n];
    }
    int integerReplacement(int n) {
        return dfs(n);
    }
};