Leetcode.397 整数替换

最新推荐文章于 2024-08-30 11:28:00 发布

感觉画质不如…原神

最新推荐文章于 2024-08-30 11:28:00 发布

阅读量28

点赞数

分类专栏： Leetcode 文章标签：算法 leetcode 贪心位运算

本文链接：https://blog.csdn.net/m0_74396439/article/details/130351768

版权

Leetcode 专栏收录该内容

441 篇文章 2 订阅

订阅专栏

题目链接

Leetcode.397 整数替换 mid

题目描述

给定一个正整数 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 <= 2^{31} - 1$

解法一：bfs

我们可以 bfs 求从起点 $n$ 到终点1 的最短距离。

时间复杂度： $O (n)$

C++代码：

class Solution {
public:
    int integerReplacement(int n) {
        //因为 n + 1 可能爆int了，所以直接把它转成 unsined int
        queue<uint> q;
        q.push(n);

        int d = 0;
        while(!q.empty()){
            int sz = q.size();
            for(int i = 0;i < sz;i++){
                auto t = q.front();
                q.pop();
                //当 t == 1 时，此时的 d 就是最少的操作次数。
                if(t == 1) return d;

                if(t & 1){
                    q.push(t - 1);
                    q.push(t + 1);
                }
                else q.push(t / 2);
            }
            d++;
        }
        return -1;
    }
};

解法二：贪心 & 位运算

当 $n$ 是偶数时， $n = n /2$
当 $n$ 是奇数时， $n = 4 k + 3 或者 n = 4 k + 1$ 。
- 当 $n = 4 k + 3$ 时，
- 4k + 3 -> 4k + 4 -> 2k + 2 -> k + 1 -> k， $n + 1$ 变为 $k$ 共 4 步；4k + 3 -> 4k + 2 -> 2k + 1 -> 2k -> k， $n - 1$ 变为 $k$ 也是共 4 步。所以当 $n = 4 k + 3$ 时， $n + 1 或者 n - 1$ 都可以。
- 当 $n = 4 k + 1$ 时，
- 4k + 1 -> 4k + 2 -> 2k + 1 -> 2k -> k， $n + 1$ 变为 $k$ 共 4 步；4k + 1 -> 4k -> 2k -> k ， $n - 1$ 变为 $k$ 共 3 步。所以当 $n = 4 k + 1$ 时，选择 $n - 1$ 更优。

例外： 当 $n = 3$ 时，选择 $n - 1$ 。

3 -> 4 -> 2 -> 1， $n + 1$ 共 3 步；3 -> 2 ->1， $n - 1$ 共 2 步。

时间复杂度： $O (l o g n)$

C++代码：

class Solution {
public:
    int integerReplacement(int n) {
        uint x = n;
        int ans = 0;

        while(x != 1){
            if((x & 1) == 0) x >>= 1;
            else{
                if(x % 4 == 1 || x == 3) x--;
                else x++;
            }
            ans++;
        }
        return ans;
    }
};