leetcode1201-丑数III

温笙

已于 2022-09-05 22:47:10 修改

阅读量279

点赞数 2

文章标签：算法 leetcode java

于 2022-09-05 22:17:42 首次发布

本文链接：https://blog.csdn.net/qq_44859565/article/details/126708952

版权

题目描述

给你四个整数：n 、a 、b 、c ，请你设计一个算法来找出第 n 个"丑数"。
"丑数"是可以被a或 b 或c 整除的正整数。
题目难度：中等
示例1

输入：n= 3, a = 2, b = 3, c = 5
输出：4
解释：丑数序列为 2, 3, 4, 5, 6, 8, 9, 10… 其中第 3 个是 4。

示例2

输入：n = 4, a = 2, b = 3, c = 4
输出：6
解释：丑数序列为 2, 3, 4, 6, 8, 9, 10, 12… 其中第 4 个是 6。

示例3

输入：n = 5, a = 2, b = 11, c = 13
输出：10
解释：丑数序列为 2, 4, 6, 8, 10, 11, 12, 13… 其中第 5 个是 10。

示例4

输入：n = 1000000000, a = 2, b = 217983653, c = 336916467
输出：1999999984

提示

1 <= a * b * c <= 10^18
1 <= n, a, b, c <= 10^9
本题结果在 [1, 2 * 10^9] 的范围内

解法一（超时）

总体思路

在本题中，容易想到的解法就是暴力破解法，其思路非常简单，实现起来也不困难。只需从a,b,c三个数字中最小的开始遍历，依次判断每一个数字是否为“丑数“并记录丑数的个数，直到遍历到第n个”丑数“为止，然后将其返回即可。

代码实现

public class solution1 {
    public static int nthUglyNumber(int n, int a, int b, int c) {
        //定义一个计数器用于记录遍历到的”丑数“个数
        int count = 0;
        //得到三个数字中最小的数字
        int i = Math.min(Math.min(a,b),c);
        while (count != n){//当当前遍历的丑数为第n个时退出循环
            if (i%a==0 || i%b==0 || i%c==0){
                count++;
            }
            if(count == n){
                break;
            }
            i++;
        }
        return i;
    }
}

运行结果

示例1运行结果
示例2运行结果
示例3运行结果
示例4运行结果

结果分析

从以上运行结果不难看出解法一能够通过所给的4个测试示例的测试，但是提交到leetcode平台上后才发现该种解法不能满足时间要求，超出时间限制。提交结果而且从结果中可以发现主要是在测试示例4时超出了时间限制，不妨简单计算一下示例4的具体运行时间。
示例4运行时间
从上图可以看出，测试示例4时运行时间约为27秒，这个确实有些难以接受，而此种解法很难再提升运行效率，至此解法一以失败告终。

解法二（二分法）

在发现解法一超时后我也积极地去寻找新的方法去降低时间复杂度，而解法一的时间复杂度为O(n)，如果再降低的话最有可能的就是O(log(n))，但是作为”小白“的我却很难去想出时间复杂度为O(log(n))的算法，无奈之下只能点开了他题目描述下方的提示。
题目提示由于提示是英文的，而我已经很久没有看过英文了，所以基本上前面大半截都看不懂，只看懂了”binary search“，由于之前学习过"二分法"查找一个有序序列中的某个值，所以对这个单词相对比较熟悉，再加上该算法的时间复杂度一般都是O(log(n))就更加坚定了我的这一想法，所以这成为了这个题目的突破口。

总体思路

所谓”二分法“就是用一个下限和一个上限确定一个区间，然后不断调整这个区间并将这个区间一分为二，直到找到符合条件的值或者下限值不小于上限值时停止。在本题中不难看出初始下限left为0，初始上限right的值为min(a,b,c)n，然后用mid记录区间的中值*，并根据**[0,mid]这个区间内的”丑数“的个数num与n的关系调整上下限值，如果num<n令left=mid+1**，如果num>n令right=mid-1，如果num==n需要判断mid是否为”丑数“，如果是丑数，返回mid即可，否则需要令right=mid-1继续查找。
从以上分析可知一个比较困难的问题就是要找到[0,mid]区间中有多少个”丑数“，这个问题如果只是靠简单地”硬想“可能很难想的明白，但是如果从集合的角度理解可能会更加清晰。具体来说就是[0,mid]区间中”丑数“的个数num=mid/a + mid/b + mid/c - mid/lcm(a,b) - mid/lcm(a,c) - mid/lcm(b,c) + mid/lcm(a,b,c)【其中lcm(a,b)是a和b的最小公倍数】，可参考下图进行理解。
在这里插入图片描述

代码实现

public class solution2 {
    public static int nthUglyNumber(int n, int a, int b, int c) {
        //初始下限为0，初始上限为三者最小数乘n
       long left = 0,right = Math.min(Math.min(a,b),c)*n;
       //当下限不小于上限时退出循环
       while (left < right){
           //找到区间中间位置
           long mid = (left + right)/2;
           //找出在mid之前有多少个”丑数“
           long num = mid/a + mid/b + mid/c - mid/lcm(a,b)  - mid/lcm(a,c) - mid/lcm(b,c) + mid/lcm(lcm(a,b),c);
           //如果在mid之前有n个丑数需要判断当前数字是否为”丑数“
           if (num == n){
               if (mid % a==0 || mid % b==0 || mid %c == 0){//如果当前位置的数字是丑数则直接返回
                   return (int) mid;
               }else {
                   right = mid-1;
               }
           }else if (num > n){//当前位置的”丑数“个数大于n
               right = mid - 1;
           }else if (num < n){//当前位置的”丑数“个数小于n
               left = mid + 1;
           }
       }
       return (int)left;
    }
    //求两个数的最小公倍数
    public static long lcm(long a, long b){
        return a/gcd(a,b)*b;
    }
    // 计算最大公约数
    private static long gcd(long a, long b) {
        return b > 0 ? gcd(b, a % b) : a;
    }
   
}

在以上代码中比较难以理解的可能就是当下限left不小于上限right退出循环时能否保证返回的left是”丑数“，其实这一点也不用想得过于复杂，只需要明白我们要找的第n个丑数一定在[left,right]这个区间内即可，但是现在left已经等于right了，相当于在这个区间内只有这一个值了，理所当然地这个值就一定是它了。（不妨拿一组测试用例n=3，a=2，b=3，c=5进行尝试）
在以上代码中需要注意的就是最小公倍数的求法，两个数最小公倍数=一个数/两个数的最大公因数*另一个数，两个数的最大公因数使用的是辗转相除法，经过提交代码证明在求最小公倍数或最大公因数时使用普通方法在测试样例4时仍然会超时，因为那样又把时间复杂度变成了O(n)，所以只能使用更加高效的方法以降低时间复杂度。