1. 題目

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

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

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/get-kth-magic-number-lcci
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

2. 解題

2.1 set “隊列”

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

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 動態規劃

• 參考
  LeetCode 263. 丑數 && 264. 丑數 II（DP）
  LeetCode 313. 超級丑數（動態規劃）
  LeetCode 1201. 丑數 III（最小公倍數+二分查找）

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

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

總結

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

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。