AI刷题 小U的数字插入问题 | 豆包MarsCode AI刷题

41 阅读3分钟

问题描述

小U手中有两个数字 a 和 b。第一个数字是一个任意的正整数,而第二个数字是一个非负整数。她的任务是将第二个数字 b 插入到第一个数字 a 的某个位置,以形成一个最大的可能数字。

你需要帮助小U找到这个插入位置,输出插入后的最大结果。


测试样例

样例1:

输入:a = 76543, b = 4
输出:765443

样例2:

输入:a = 1, b = 0
输出:10

样例3:

输入:a = 44, b = 5
输出:544

样例4:

输入:a = 666, b = 6
输出:6666

一、解题思路

  1. 转换为字符串处理

    • 由于我们要将一个数字插入到另一个数字的某个位置以得到最大的数,将数字转换为字符串处理会更方便。这样我们可以在字符串的不同位置插入另一个数字的字符串表示,然后再比较得到的结果。
  2. 遍历插入位置

    • 我们需要遍历第一个数字的字符串表示的所有可能插入位置。对于每一个位置,将第二个数字的字符串插入到该位置,然后将得到的新字符串转换回数字,比较这些数字的大小,找到最大的那个。

下面以java为例进行代码实现:

public class Main {
    public static int solution(int a, int b) {
        String aStr = String.valueOf(a);
        String bStr = String.valueOf(b);
        int max = 0;
        for (int i = 0; i <= aStr.length(); i++) {
            String newStr = aStr.substring(0, i)+bStr + aStr.substring(i);
            int num = Integer.parseInt (newStr);
            if (num > max) {
                max = num;
            }
        }
        return max;
    }

    public static void main(String[] args) {
        System.out.println(solution(76543, 4) == 765443);
        System.out.println(solution(1, 0) == 10);
        System.out.println(solution(44, 5) == 544);
        System.out.println(solution(666, 6) == 6666);
    }
}

二、代码解读

  1. 函数定义与参数接收

    • 定义了一个名为insertToMax的静态函数,它接受两个整型参数ab。其中a是任意的正整数,b是非负整数。
  2. 初始化与循环

    • 在函数内部,首先将ab转换为字符串aStrbStr。然后定义一个变量max并初始化为0,用于存储最大的结果。接着通过一个for循环,循环变量i从0到aStr.length(),这个循环用于遍历a的字符串表示的所有可能插入位置。
  3. 字符串拼接与转换

    • 在循环内部,对于每一个i,将bStr插入到aStr的第i个位置,得到新的字符串newStr。然后将newStr转换回整型数字num
  4. 比较与更新最大值

    • 如果num大于max,则将max更新为num
  5. 结果返回

    • 最后,当循环结束后,返回max,这个值就是将b插入到a中某个位置后得到的最大数字。

三、时间复杂度和空间复杂度分析

  1. 时间复杂度

    • 在最坏的情况下,我们需要遍历第一个数字a的字符串表示的每一个位置,假设a的数字长度为n,那么时间复杂度为O(n)O(n)。
  2. 空间复杂度

    • 我们创建了一些临时的字符串和变量,其中占用空间最多的是存储ab的字符串表示。假设a的数字长度为nb的数字长度为m,那么空间复杂度为O(n+m)O(n+m),因为我们存储了aStrbStr,但通常mn相对较小,并且不会随着输入规模的增长而呈指数级增长。

四、总结

这道题目比较简单,使用循环遍历字符串就可以了。当然也肯定有复杂度更低的算法。