每日一道Leetcode - 面试题 17.09. 第 k 个数 [动态规划｜三指针]

奋进的杰西

于 2021-07-07 10:39:52 发布

阅读量134

点赞数

分类专栏： Leetcode 文章标签：指针

本文链接：https://blog.youkuaiyun.com/weixin_41041275/article/details/118541760

版权

Leetcode 专栏收录该内容

304 篇文章

订阅专栏

该博客介绍了如何使用动态规划和三指针算法找到第k个魔法数。魔法数是3、5或7的倍数，通过维护三个指针分别遍历3、5、7的倍数列表，找到最小的魔法数并更新状态。最终，返回dp[k-1]即为第k个魔法数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述

class Solution:
    def getKthMagicNumber(self, k: int) -> int:
        dp = [0]*(k+1)
        dp[0] = 1
        # 三个指针分别指向三个列表头【第一个是以3为倍数的列表，第二个是以5为倍数的列表，第三个是以7为倍数的列表，在这三个数字中找最小的，再滑动指针】
        l1,l2,l3 = 0,0,0
        for i in range(1,k):
            dp[i] = min(dp[l1]*3,dp[l2]*5,dp[l3]*7)
            if dp[i] == dp[l1] * 3:
                l1 += 1
            if dp[i] == dp[l2] * 5:
                l2 += 1
            if dp[i] == dp[l3] * 7:
                l3 += 1
        return dp[k-1]