面試題34. 醜數
題目描述
把隻包含因子2、3和5的數稱作醜數(Ugly Number)。例如6、8都是醜數,但14不是,因為它包含因子7。
習慣上我們把1當做是第一個醜數。求按從小到大的順序的第N個醜數。
思路:
醜數應該是另一個醜數乘以2、3、5得到的。也就是說,根據一個醜數,可以計算出它乘以2、3、5得到的醜數。是以,可以初始化一個數組arr,用于儲存已經找到的醜數。然後按序用數組裡的數,乘以2、3、5得到新的醜數,并加入到數組中。
public class Solution
public int GetUglyNumber_Solution(int n) {
if(n <= 0) return 0;
int[] arr = new int[n];
arr[0] = 1;
int i = 0, j = 0, k = 0; // 分别代表2,3,5的下标
for(int a = 1; a < n; a++) {
int p = arr[i] << 1;
int q = arr[j] * 3;
int r = arr[k] * 5;
int min = Math.min(p, Math.min(q, r));
arr[a] = min;
if(p == min) {
i++;
}
if(q == min) {
j++;
}
if(r == min) {
k++;
}
}
return arr[n-1];
}
}