leetcode 397. Integer Replacement

最新推荐文章于 2025-08-11 15:06:51 发布

原创最新推荐文章于 2025-08-11 15:06:51 发布 · 349 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #397

Leetcode 专栏收录该内容

175 篇文章

订阅专栏

本文探讨了在解决特定问题时广度优先搜索算法的一种优化方法。通过避免使用耗时的操作如cout，并调整数据类型来处理大整数，显著提高了算法效率。文中提供了优化前后的代码实现对比。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

开始想的是采用类似于广度搜索图的算法计算，发现它超时了。

超时的版本如下：

class Solution {
public:
    typedef struct trace
    {
        int num;
        int residual;
    }TRACE;

    int integerReplacement(int n) {
        int rel = INT_MAX;
        int ini = 0;
        queue<TRACE> remain;

        cout << rel << endl; 

        while(1)
        {
            if(!ini)
            {
                if(n == 1)
                    return 0;

                if(n&1)
                {
                    TRACE t_h = {n+1, 1};
                    TRACE t_l = {n-1, 1};
                    remain.push(t_h);
                    remain.push(t_l);
                }    
                else
                {
                    TRACE t = {n >> 1, 1};
                    remain.push(t);
                }

                ini = 1;
            }
            else
            {
                int cur_size = remain.size();
                cout << cur_size << endl;

                for(int i = 0; i < cur_size; i++)
                {
                    TRACE t_top = remain.front();
                    remain.pop();
                    cout << t_top.num << "  " << t_top.residual << endl;
                    if(t_top.num == 1 && t_top.residual < rel)
                    {
                        rel = t_top.residual;
                        return rel;
                    }    
                    else if(t_top.num & 1)
                    {
                        TRACE t_h = {t_top.num+1, t_top.residual+1};
                        TRACE t_l = {t_top.num-1, t_top.residual+1}; 
                        remain.push(t_h);
                        remain.push(t_l);                        
                    }   
                    else
                    {
                       TRACE t = {t_top.num >> 1, t_top.residual+1};
                       remain.push(t);
                    }
                }

                if(remain.empty())
                    return rel;
            }
        }

        return rel;
    }
};

后来我觉得这里面可能有问题，原来是cout占了不少时间。后来我运行，发现输入的n居然有可能是最大的有符号的正整数，所以我进行一个类型转换。下面就是可以的代码。

class Solution {
public:
    typedef struct trace
    {
        unsigned int num;
        int residual;
    }TRACE;

    int integerReplacement(int n) {
        int rel = INT_MAX;
        int ini = 0;
        queue<TRACE> remain;

        cout << rel << endl; 

        while(1)
        {
            if(!ini)
            {
                if(n == 1)
                    return 0;

                if(n&1)
                {
                    TRACE t_h = {(unsigned int)(n+1), 1};
                    remain.push(t_h);
                    TRACE t_l = {n-1, 1};
                    remain.push(t_l);
                }    
                else
                {
                    TRACE t = {n >> 1, 1};
                    remain.push(t);
                }

                ini = 1;
            }
            else
            {
                int cur_size = remain.size();
                // cout << cur_size << endl;

                for(int i = 0; i < cur_size; i++)
                {
                    TRACE t_top = remain.front();
                    remain.pop();
                    // cout << t_top.num << "  " << t_top.residual << endl;
                    if(t_top.num == 1 && t_top.residual < rel)
                    {
                        rel = t_top.residual;
                        return rel;
                    }    
                    else if(t_top.num & 1)
                    {
                        TRACE t_h = {t_top.num+1, t_top.residual+1};
                        TRACE t_l = {t_top.num-1, t_top.residual+1}; 
                        remain.push(t_h);
                        remain.push(t_l);                        
                    }   
                    else
                    {
                       int tmp = t_top.num >> 1;
                       TRACE t = {tmp, t_top.residual+1};
                       remain.push(t);
                    }
                }

                if(remain.empty())
                    return rel;
            }
        }

        return rel;
    }
};