剑指 Offer 49. 丑数（待补充）

剑指 Offer 49. 丑数 - 力扣（LeetCode）

目录

分析

代码

---

分析

//假设当前已经获得一段丑数序列，M为当前丑数序列的最大值
//我们要扩大该序列,即求出下一个丑数
// 该序列中存在三个特殊的数：m2, m3, m5
// m2满足：在序列中，所有小于m2的数*2均<=M，所有大于m2的数*2均>M
// 根据m2,m3,m5我们可求出M2,M3,M5，M2=m2*2, M3=m3*3, M5=m5*5
// 则下一个丑数即为min(M2, M3, M5)
// 
// 因此只要我们有一段排列好的丑数序列，就可以依照这种方式来无限扩展
// 扩展之后新的m2,m3,m5一定大于原来的值，只需要逐个向后检查即可

代码

class Solution {
public:
    int nthUglyNumber(int n) {
        vector<int> a(n);
        a[0] = 1;
        int p2 = 0, p3 = 0, p5 = 0;
        int M = 0;
        while (--n) {
            while (a[p2] * 2 <= a[M]) ++p2;
            while (a[p3] * 3 <= a[M])++p3;
            while (a[p5] * 5 <= a[M])++p5;
            a[++M] = Min(a[p2] * 2, a[p3] * 3, a[p5] * 5);
        }
        return a.back();
    }

    int Min(int a, int b, int c) {
        int m1 = a < b ? a : b;
        int m2 = m1 < c ? m1 : c;
        return m2;
    }
};