Leetcode - Ugly Number II

[分析]
暴力的办法就是从1开始检查每个数是否是丑数,发现丑数计数器加1直到找到第n个丑数。这种方法效率低,因为它不管是不是丑数都进行了计算,所以优化方向是仅计算丑数而不在非丑数上耗费时间。易知后面的丑数一定是前面的丑数乘以2或者3或者5得到的,假设现在已经计算出第k个丑数U(k),那么下一个丑数是前面丑数中乘以2、3、5中第一个大于U(k)的数。怎么找呢?我们需要保留已计算的所有丑数,将已知丑数逐个乘2并同U(k)比较,第一个大于U(k)的数记为next2, 同样的方式计算出next3, next5, 下一个丑数就是next2, next3, next5三者间的最小值。每次必须从第一个丑数开始找吗?不是的,可以进一步优化,对乘以2而言,肯定存在某个丑数T2, 其前的丑数乘2小于U(k), 其以后的都大于U(k), 记录T2所在位置,每次生成新丑数时更新该位置,对乘以3和乘以5同样记录相应位置。
[ref]
[url]https://leetcode.com/discuss/52716/o-n-java-solution[/url]

public class Solution {
public int nthUglyNumber(int n) {
int[] uglyNum = new int[n];
uglyNum[0] = 1;
int p2 = 0, p3 = 0, p5 = 0;
int next2 = 2, next3 = 3, next5 = 5;
for (int i = 1; i < n; i++){
uglyNum[i] = Math.min(Math.min(next2, next3), next5);
if (uglyNum[i] == next2) {
next2 = uglyNum[++p2] * 2;
}
if (uglyNum[i] == next3) {
next3 = uglyNum[++p3] * 3;
}
if (uglyNum[i] == next5) {
next5 = uglyNum[++p5] * 5;
}
}
return uglyNum[n-1];
}
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值