思路:做题之前,要知道什么是丑数,按照百度百科的说法是,1和2,3,5他们的倍数x属于1到无穷大,他们的集合就是{1,5*x,2*x,3*x},有一点丑数必须由一个丑数和另一个因子相乘得到,如7,14不是丑数
丑数的性质: 丑数只包含因子 2, 3, 5,因此有 “丑数 == 某较小丑数 × 某因子” (例如:10 =5×2)。
得出这么些丑数不是难点,难点是:精准的去重(举例x =3 ,5*3=15,当x=5,3*5 =15;随着遍历会得到两个15),取出第n个大的丑数
方法一:最小堆
考虑用最小堆的原因是可以用优先队列进行存放这些丑数,从小到大自动排序,在排除重复元素的情况下,第 n 次从最小堆中取出的元素即为第 n个丑数
//最小堆+优先队列
class Solution {
public:
int nthUglyNumber(int n) {
if(n==1)return 1;
double res = 1;
priority_queue<double,vector<double>,greater<double>> pq;
for(int i =1;i<n;i++){
pq.push(res*2);pq.push(res*3);pq.push(res*5);//如果,n =2存入2,3,5,头部的2即为所得
res = pq.top();
//去除头部元素
pq.pop();
//pq.erase(res);//没有这个函数
//如果剩下的头部元素和刚才去除的相同,应当一并去除
while(!pq.empty()&&res == pq.top())pq.pop();
}
return res;
}
};
其实set本来就是有序,可以直接用set来排序,而不需要再用优先队列
//使用set
class Solution {
public:
int nthUglyNumber(int n) {
set<double> s; // set 是有序的,且不重复
double answer = 1;
for (int i = 1; i < n; i++) {
s.insert(answer * 2);
s.insert(answer * 3);
s.insert(answer * 5);
answer = *s.begin(); s.erase(answer);
}
return answer;
}
};
方法二:暴力求解
话不多说直接上代码
//暴力
class Solution {
public:
int nthUglyNumber(int n) {
vector<int> v;
for (long long a=1;a<=INT_MAX;a=a*2)
for (long long b=a;b<=INT_MAX;b=b*3)
for (long long c=b;c<=INT_MAX;c=c*5)
v.push_back(c);
sort(v.begin(),v.end());
return v.at(n-1);
}
};
方法三:动态规划(三指针)
/动态规划
class Solution {
public:
int nthUglyNumber(int n) {
//vector<int> dp(n,0);用这个会浪费更大空间,不如直接用静态数组
int dp[n];
int a =0, b =0, c=0;
dp[0] = 1;
for(int i=1;i<n;i++){
int dpa = dp[a]*2, dpb = dp[b]*3,dpc = dp[c]*5;
dp[i] = min(min(dpa,dpb),dpc);
if(dp[i]==dpa)a++;
if(dp[i]==dpb)b++;
if(dp[i]==dpc)c++;
}
return dp[n-1];
}
};