面试题 17.09. 第 k 个数
有些数的素因子只有 3,5,7,请设计一个算法找出第 k 个数。注意,不是必须有这些素因子,而是必须不包含其他的素因子。例如,前几个数按顺序应该是 1,3,5,7,9,15,21。
示例 1:
输入: k = 5
输出: 9
堆+(bfs?)
- 很容易想到堆,初始堆顶为
1
, - 每次弹出堆顶
top
时,就把top*3
,top*5
和top*7
压入堆, - 直到
k
个不重复的堆顶弹出即可得到答案。注意去重
动态规划,三指针
//堆bfs
int getKthMagicNumber(int k) {
priority_queue<ll, vector<ll>, greater<ll> > q;
ll ans = 0, a[4] = { 3, 5, 7 };
q.push(1);
unordered_set<int> vis;
while(k) {
ans = q.top(); q.pop();
vis.insert(ans);
k --;
for(int i=0; i<3; i++) {
if(vis.count(ans*a[i])) continue ;
vis.insert(ans*a[i]);
q.push(ans*a[i]);
}
}
return ans;
}
//3指针
int getKthMagicNumber(int k) {
ll dp[k+1] = { 0 };
dp[0] = 1;
ll i3 = 0, i5 = 0, i7 = 0;
for(int i=1; i<k; i++) {
dp[i] = min({dp[i3]*3, dp[i5]*5, dp[i7]*7});
if(dp[i] == dp[i3]*3) i3 ++;
if(dp[i] == dp[i5]*5) i5 ++;
if(dp[i] == dp[i7]*7) i7 ++;
}
return dp[k-1];
}