题目描述
把只包含质因子2、3和5的数称作丑数(Ugly Number)。例如6、8都是丑数,但14不是,因为它包含质因子7。 习惯上我们把1当做是第一个丑数。求按从小到大的顺序的第N个丑数。
之前做题都是怎么想就怎么做,根本不考虑性能方面的问题。但是这个题造成运行超时,不得不考虑了。
先放错误代码:
public class Solution {
public int GetUglyNumber_Solution(int index) {
if(index == 0)
return 0;
int count=0,sum=2,sum1;
while(true){
sum1 = sum;
while(sum1%2 == 0){
sum1=sum1/2;
}
while(sum1%3 == 0){
sum1=sum1/2;
}
while(sum1%5 == 0){
sum1=sum1/2;
}
if(sum1 == 1){
count++;
if(count == index)
break;
}
sum++;
}
return sum;
}
}
那么正确的代码实现是:
import java.util.*;
public class Solution {
public int GetUglyNumber_Solution(int index) {
if(index == 0)
return 0;
int[] result = new int[index];
int count = 0;
int i2 = 0;
int i3 = 0;
int i5 = 0;
result[0] = 1;
int tmp = 0;
while (count < index-1) {
tmp = Math.min(result[i2] * 2, Math.min(result[i3] * 3, result[i5] * 5));
if(tmp==result[i2] * 2)
i2++;
if(tmp==result[i3] * 3)
i3++;
if(tmp==result[i5]*5)
i5++;
result[++count]=tmp;
}
return result[index - 1];
}
}