算法之隊列求第 k 個數

問題描述

有些數的素因子只有 3，5，7，請設計一個算法找出第 k 個數。注意，不是必須有這些素因子，而是必須不包含其他的素因子。例如，前幾個數按順序應該是 1，3，5，7，9，15，21。

示例 1:
輸入: k = 5
輸出: 9

算法實現

class Solution {
	/**
	* 醜數
	* 通過三個索引記錄最小值
	*/
    public int getKthMagicNumber(int k) {
        int[] arr = new int[k];
        arr[0] = 1;
        int pos3=0,pos5=0,pos7=0;
        for(int i=1;i<k;i++){
            int current;
            if(arr[pos3]*3<arr[pos5]*5){
                current = arr[pos3]*3;
            }else{
                current = arr[pos5]*5;
            }
            if(current > arr[pos7]*7){
                current = arr[pos7]*7;
            }
            if(current==arr[pos3]*3){
                pos3++;
            }
            if(current==arr[pos5]*5){
                pos5++;
            }
            if(current==arr[pos7]*7){
                pos7++;
            }
            arr[i]=current;
        }
        return arr[k-1];
    }
}