有些数的素因子只有 3,5,7,请设计一个算法找出第 k 个数。注意,不是必须有这些素因子,而是必须不包含其他的素因子。例如,前几个数按顺序应该是 1,3,5,7,9,15,21。
示例 1:
输入: k = 5
输出: 9
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/get-kth-magic-number-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:三指针法,首先将p3,p5,p7
移动到合适的位置(p3*3
大于上一个丑数的第一个位置,p5*5
大于上一个丑数的第一个位置,p7*7
大于上一个丑数的第一个位置),此时指针已经就绪,那么下一个丑数就是这些结果中的最小值,因为任意一个丑数必然是前面的丑数乘3
,乘5
,乘7
得到的。
具体解析可以看评论区大佬的,其实和动态规划是有关系的。
class Solution {
public:
int getKthMagicNumber(int k) {
vector<int> res(k);
int p3 = 0,p5 = 0, p7 = 0;
res[0] = 1;
int lastUgly = 0;
for(int i = 1;i < k;i++)
{
lastUgly = res[i - 1];
while(lastUgly>= res[p3] * 3)//将以3为因子的最小丑数就位
p3++;
while(lastUgly>= res[p5] * 5)//将以5为因子的最小丑数就位
p5++;
while(lastUgly>= res[p7] * 7)//将以7为因子的最小丑数就位
p7++;
res[i] = min(min(res[p3] * 3,res[p5] * 5),res[p7]*7);
}
return res[k - 1];
}
};