剑指 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;
}
};