1. 題目

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

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

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/get-kth-magic-number-lcci
著作權(quán)歸領(lǐng)扣網(wǎng)絡(luò)所有。商業(yè)轉(zhuǎn)載請聯(lián)系官方授權(quán)，非商業(yè)轉(zhuǎn)載請注明出處。

2. 解題

2.1 set “隊列”

• 滿足題目要求的數(shù)只能乘以3、5、7
• 用set，有序，可以去重、當(dāng)做優(yōu)先隊列
• 不斷的出隊begin(), 且把 begin() 的 3,5,7倍數(shù)插入隊列

class Solution { public:int getKthMagicNumber(int k) {set<long> q;//可以看做小頂堆long ans;q.insert(1);while(k--){ans = *q.begin();q.erase(q.begin());q.insert(ans*3);q.insert(ans*5);q.insert(ans*7);}return ans;} };

2.2 動態(tài)規(guī)劃

• 參考
  LeetCode 263. 丑數(shù) && 264. 丑數(shù) II（DP）
  LeetCode 313. 超級丑數(shù)（動態(tài)規(guī)劃）
  LeetCode 1201. 丑數(shù) III（最小公倍數(shù)+二分查找）

自己畫一下就明白了，比較潦草，見諒

class Solution { public:int getKthMagicNumber(int k) {vector<long> dp(k+1,0);dp[1] = 1;int i3=1, i5=1, i7=1;for(int i = 2; i <= k; i++){dp[i] = min(dp[i3]*3, min(dp[i5]*5, dp[i7]*7));if(dp[i3]*3 == dp[i])i3++;if(dp[i5]*5 == dp[i])i5++;if(dp[i7]*7 == dp[i])i7++;}return dp[k];} };

0 ms 6.1 MB

總結(jié)

以上是生活随笔為你收集整理的程序员面试金典 - 面试题 17.09. 第 k 个数（set优先队列/DP）的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。