LeetCode每日一题——面试题 17.09. 第 k 个数

LeetCode每日一题系列

题目：面试题 17.09. 第 k 个数
难度：普通

---

---

题目

有些数的素因子只有 3，5，7，请设计一个算法找出第 k 个数。注意，不是必须有这些素因子，而是必须不包含其他的素因子。例如，前几个数按顺序应该是 1，3，5，7，9，15，21。

示例

示例 1:

输入: k = 5

输出: 9

思路

动态规划：题目要求素因子只为3,5,7的数，我们干脆直接就按顺序从1开始一直用[3,5,7]的数相互相乘，前几个数即为1,13，15,17,133,135，137,153,155,15*7…等等。想到这已经可以使用动态规划了，这里注意从前往后乘的时候可能会出现重复值，我们需要祛除重复值并且保持答案的相对位置不变。
这里使用了index1，index2分别代表[3,5,7]的坐标与tmp中我们需要相乘因子的坐标（从第一个数开始，每乘三次加一）

题解

class Solution:
    def getKthMagicNumber(self, k: int) -> int:
        tmp = [1]
        res = [3,5,7]
        index1, index2 = 0, 0
        for i in range(1, 4000):
            # index2加一
            if i != 1 and (i - 1) % 3 == 0:
                index2 += 1
            # 去重
            if tmp[index2] * res[index1] not in tmp:
                tmp.append(tmp[index2] * res[index1])
            # index每乘一次都要加一对3取余
            index1 = (index1 + 1) % 3
        # 排序以保持相对位置
        tmp = sorted(list(tmp)
        # 返回下标为k-1的值
        return tmp[k-1]