数字增值问题| 豆包MarsCode AI 刷题

237 阅读3分钟

问题描述

给定一个正整数 n,定义一次"增殖"操作如下:

  • 将数字 n 转化为一个包含 1 到 n 的递增序列
  • 例如:当 n = 4 时,一次增殖后变为序列 [1, 2, 3, 4]

现给定三个正整数:

  • 初始数字 n
  • 增殖次数 k
  • 位置索引 p(从1开始计数)

请计算经过 k 次增殖操作后,序列中第 p 个位置的数字。如果 p 超出序列长度,则返回 -1

示例说明

示例 1:

测试样例

样例1:

输入:n = 4 ,k = 3 ,p = 19
输出:3
解释:
第1次增殖:4 -> [1,2,3,4]
第2次增殖:[1,2,3,4] -> [1, 1,2, 1,2,3, 1,2,3,4]
第3次增殖:得到最终序列 [1, 1, 1,2, 1, 1,2, 1,2,3, 1, 1,2, 1,2,3, 1,2,(3),4] -> ,第19个位置的数字是3

样例2:

输入:n = 3 ,k = 2 ,p = 5
输出:2
解释:
第1次增殖:3 -> [1,2,3]
第2次增殖:[1,2,3] -> [1, 1,2, 1,(2),3]
第5个位置的数字是2

样例3:

输入:n = 5 ,k = 1 ,p = 7
输出:-1
解释:
第1次增殖:5 -> [1,2,3,4,5]
序列长度为5,p=7超出范围,返回-1

这道题目主要考察对序列生成规则的理解以及如何高效地定位某个位置的数字。以下是题解:


题目分析

  1. 增殖操作规则

    • 初始数字 nn 转换为序列 [1,2,…,n][1, 2, \dots, n]。
    • 每次增殖后,将当前序列中的每个数字 xx 替换为 [1,2,…,x][1, 2, \dots, x]。
    • 每次增殖操作会显著增加序列长度。
  2. 问题关键

    • 随着增殖次数 kk 的增加,序列长度呈指数增长,直接模拟构造序列会导致性能问题。
    • 需要通过数学方法直接定位 pp 的位置,避免完整生成序列。

解题思路

  1. 计算序列长度

    • 假设初始数字为 nn,经过 kk 次增殖后的序列长度可以递归计算: L(0)=1,L(k)=∑i=1nL(k−1,i)L(0) = 1, \quad L(k) = \sum_{i=1}^{n} L(k-1, i)
    • 对于每个数字 xx,其子序列长度与 xx 成比例。
  2. 定位第 pp 个数字

    • 利用递归,根据 pp 的值逐步缩小范围,无需生成完整序列。
    • 从 kk 次增殖操作开始,判断 pp 所属的子序列,然后递归查找。
  3. 特殊情况

    • 若 pp 超过当前序列的长度,则直接返回 -1。

实现代码

以下为具体实现:

def find_value(n, k, p):
    # 计算序列长度
    def sequence_length(n, k):
        if k == 0:
            return 1
        lengths = [sequence_length(i, k - 1) for i in range(1, n + 1)]
        return sum(lengths)

    # 定位 p 的数字
    def find_position(n, k, p):
        if k == 0:
            return n if p == 1 else -1
        
        cumulative_length = 0
        for i in range(1, n + 1):
            # 当前子序列的长度
            current_length = sequence_length(i, k - 1)
            if cumulative_length + current_length >= p:
                # 在当前子序列中递归寻找
                return find_position(i, k - 1, p - cumulative_length)
            cumulative_length += current_length
        
        return -1  # 超出范围

    # 计算第 p 个位置的值
    return find_position(n, k, p)

# 测试样例
print(find_value(4, 3, 19))  # 输出:3
print(find_value(3, 2, 5))   # 输出:2
print(find_value(5, 1, 7))   # 输出:-1

复杂度分析

  1. 时间复杂度

    • 每次递归会在当前层检查所有数字,最坏情况下需要递归 kk 层: O(nk)O(n^k)
    • 通过缓存优化(如动态规划),可以显著减少重复计算。
  2. 空间复杂度

    • 主要是递归调用的栈空间,复杂度为 O(k)O(k)。

题目总结

本题通过递归和数学推导,避免了直接构造大规模序列的问题,是一道经典的动态规划和递归结合的题目。掌握此类题目的关键在于找到规律,逐步缩小问题规模。